yujiwang0606 commited on
Commit
797f4e1
·
1 Parent(s): 1a003c2
Files changed (1) hide show
  1. wan2pt1_t2v_rcm_infer.py +0 -1
wan2pt1_t2v_rcm_infer.py CHANGED
@@ -132,7 +132,6 @@ def prepare_models(dit_path_1p3B, dit_path_14B, vae_path, text_encoder_path):
132
 
133
  net_1p3B = load_dit_model(net_1p3B, dit_path_1p3B)
134
  net_14B = load_dit_model(net_14B, dit_path_14B)
135
- net_14B = None
136
 
137
  tokenizer = Wan2pt1VAEInterface(vae_pth=vae_path)
138
  t5_encoder = UMT5EncoderModel(text_len=512, device="cuda", checkpoint_path=text_encoder_path)
 
132
 
133
  net_1p3B = load_dit_model(net_1p3B, dit_path_1p3B)
134
  net_14B = load_dit_model(net_14B, dit_path_14B)
 
135
 
136
  tokenizer = Wan2pt1VAEInterface(vae_pth=vae_path)
137
  t5_encoder = UMT5EncoderModel(text_len=512, device="cuda", checkpoint_path=text_encoder_path)