IMELE icon indicating copy to clipboard operation
IMELE copied to clipboard

The weight tar file is broken?

Open Lost-little-dinosaur opened this issue 1 year ago • 1 comments

I download the The weight tar file from https://drive.google.com/file/d/1KZ50MQY5Fof8SAoJN34bxnk7mfou4frF/view?usp=sharing

But it said that the file has error and I can't decompress it...

Please help...

Lost-little-dinosaur avatar Sep 11 '24 11:09 Lost-little-dinosaur

@Lost-little-dinosaur Don't decompress. You can directly load the model.

state_dict = torch.load('Block0_skip_model_110.pth.tar', weights_only=True)['state_dict']
model.load_state_dict(state_dict, strict=False)

VaasuDevanS avatar Oct 23 '24 12:10 VaasuDevanS