search

yuval-alaluf / SAM

Official Implementation for "Only a Matter of Style: Age Transformation Using a Style-Based Regression Model" (SIGGRAPH 2021) https://arxiv.org/abs/2102.02754
https://yuval-alaluf.github.io/SAM/
MIT License
632 stars 151 forks source link

UnpicklingError: invalid load key, '<'. #74

Closed gam-bit closed 2 months ago

gam-bit commented 2 months ago

image

I was running the notebook in Colab.

In Step 3: Load Pretrained Model of the Inference Notebook and Animation Notebook, an error similar to the one in the image occurs.

I tried changing the torch version by running:

!pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0 --extra-index-url https://download.pytorch.org/whl/cu113

but the same error occurs.

I read that such an error can occur if the model is corrupted. Could the model be damaged? Or do you have any other suggestions or reasons why this might be happening? Your advice would be greatly appreciated.

gam-bit commented 2 months ago

It works when I download the model directly and load it!