salesforce / progen

Official release of the ProGen models

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Segmentation fault (core dumped)

lvdongyuan opened this issue · comments

Hello,
when I run sample.py on ubuntu wsl of windows, it appears like this:
:~/progen-main/progen2$ python3 sample.py --model ${model} --t 0.8 --p 0.9 --max-length 1024 --num-samples 2 --context "1"
falling back to cpu
falling back to fp32
loading parameters
Segmentation fault (core dumped)
What's wrong with it? I also can't dowload torch==1.9.0. I fixed the requirements.txt as follows and it can be continue.
GNU nano 6.2 requirements.txt --find-links https://download.pytorch.org/whl/torch_stable.html
torch
transformers
tokenizers