search

google-deepmind / gemma

Open weights LLM from Google DeepMind.
http://ai.google.dev/gemma
Apache License 2.0
2.22k stars 269 forks source link

Issue when "Running the unit tests" #23

Open hedaodao5 opened 3 months ago

hedaodao5 commented 3 months ago

when use pytest . one test failed: FAILED gemma/positional_embeddings_test.py::PositionalEmbeddingsTest::test_adds_positional_embeddings0 - ValueError: Operands could not be broadcast together for add on shapes (2, 1, 1 , 5) (5,) and with the config option jax_numpy_rank_promotion='raise'. For more information, see htt...

AND when I running Sampling.py, it failed ,stop at the last ste image

IS there any relationship between them?OR other reason of the sampling failure? My JAX version is CPU,WIN 10 system,cuda 12.1,jax==0.25.0