Error in shape mismatch
#26
by
tianleliphoebe
- opened
get an error: RuntimeError: shape '[1, 526, 3072]' is invalid for input of size 2154496 in modeling_gemma.py
Hi there! Please upgrade torch as per https://huggingface.co/google/gemma-7b/discussions/17#65d64073550694a180cfd5df
osanseviero
changed discussion status to
closed