-
Notifications
You must be signed in to change notification settings - Fork 54
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Error running benchmarks/benchmark_generation.py #3
Comments
fixes HazyResearch#3 when pykeops is not installed atleast
@BlinkDL try with this fix now! For the KeOps issue - can you share details about your environment? PyTorch, CUDA, and KeOps versions would all be helpful. |
I believe the issue happens when the GPU runs out of mem for the larger benchmarks.... e.g. on my setup with
but works if i do a smaller test:
hope that helps! |
getting this same error -- also on a 24gb card. i see the --ckpt option but is there some way to toggle the model architecture between the different model sizes (ex. 1.3B vs. 2.7B) thanks! |
There are examples to switch between the different models for text generation: https://github.com/HazyResearch/H3/tree/main/examples |
Hi there. It's great to see another LM trained on the Pile.
When I run benchmarks/benchmark_generation.py:
and it exits after "Segmentation fault".
So I uninstall pykeops, and then the new error is:
The text was updated successfully, but these errors were encountered: