Skip to content

GitLab

Commit fab43a07 authored by gushiqiao's avatar gushiqiao Committed by GitHub
Browse files

Update vae_2_2.py

parent 7516ad2a
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
lightx2v/models/video_encoders/hf/wan/vae_2_2.py
View file @ fab43a07
...@@ -828,7 +828,7 @@ def _video_vae(pretrained_path=None, z_dim=16, dim=160, device="cpu", cpu_offloa ...@@ -828,7 +828,7 @@ def _video_vae(pretrained_path=None, z_dim=16, dim=160, device="cpu", cpu_offloa
# load checkpoint # load checkpoint
logging.info(f"loading {pretrained_path}") logging.info(f"loading {pretrained_path}")
weights_dict = load_weights(pretrained_path, cpu_offload=cpu_offload) weights_dict = load_weights(pretrained_path, cpu_offload=cpu_offload)
model.load_state_dict(weights_dict) model.load_state_dict(weights_dict, assign=True)
return model return model
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment