Spaces:
Running
Running
Update requirements and subsequent model.pkl
#5
by
evylz
- opened
Saw that the space is having issues running.
I cloned locally and had to change the environment to train and run the gradio demo on a computer with an Nvidia GPU.
Did not need to change any of the code in the notebooks.
evylz
changed pull request status to
open
This should resolve the Runtime Error
Traceback (most recent call last):
File "/home/user/app/app.py", line 11, in <module>
learn = load_learner('model.pkl')
File "/home/user/.local/lib/python3.10/site-packages/fastai/learner.py", line 446, in load_learner
try: res = torch.load(fname, map_location=map_loc, pickle_module=pickle_module)
File "/home/user/.local/lib/python3.10/site-packages/torch/serialization.py", line 712, in load
return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args)
File "/home/user/.local/lib/python3.10/site-packages/torch/serialization.py", line 1046, in _load
result = unpickler.load()
File "/home/user/.local/lib/python3.10/site-packages/torch/serialization.py", line 1039, in find_class
return super().find_class(mod_name, name)
ModuleNotFoundError: No module named 'timm.models.layers.mlp'
jph00
changed pull request status to
merged
Many thanks!