Open-Sora icon indicating copy to clipboard operation
Open-Sora copied to clipboard

Published 20 hours ago • verified publisher icon hpcaitech

How to load Open-Sora 1.1 checkpoint weights

Open simonfei123 opened this issue 9 months ago • 1 comments

I know it can be downloaded from https://huggingface.co/hpcai-tech/OpenSora-STDiT-v2-stage2/tree/main but before there was only a .pth file and I can just save the file to ./pretrained_models. But now saving the model.safetensors doesn't work.

simonfei123 avatar May 02 '24 00:05 simonfei123

#322

hadipash avatar May 02 '24 02:05 hadipash

Now we have added automatical checkpoint loading.

zhengzangw avatar May 09 '24 09:05 zhengzangw