diff --git a/opensora/models/vae/vae.py b/opensora/models/vae/vae.py index 4a9a278..a4637e0 100644 --- a/opensora/models/vae/vae.py +++ b/opensora/models/vae/vae.py @@ -253,6 +253,7 @@ def OpenSoraVAE_V1_2( local_files_only=False, freeze_vae_2d=False, cal_loss=False, + force_huggingface=True, ): vae_2d = dict( type="VideoAutoencoderKL", @@ -277,7 +278,7 @@ def OpenSoraVAE_V1_2( scale=scale, ) - if from_pretrained is not None and not os.path.isdir(from_pretrained): + if force_huggingface or (from_pretrained is not None and not os.path.isdir(from_pretrained)): model = VideoAutoencoderPipeline.from_pretrained(from_pretrained, **kwargs) else: config = VideoAutoencoderPipelineConfig(**kwargs)