alexnasa commited on
Commit
aa0dca3
·
verified ·
1 Parent(s): c97d42c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -8
app.py CHANGED
@@ -188,19 +188,12 @@ class WanInferencePipeline(nn.Module):
188
  model_manager.load_models(
189
  [
190
  self.args.dit_path.split(","),
 
191
  self.args.text_encoder_path
192
  ],
193
  torch_dtype=self.dtype,
194
  device='cuda',
195
  )
196
-
197
- model_manager.load_models(
198
- [
199
- self.args.vae_path,
200
- ],
201
- torch_dtype=torch.float32,
202
- device='cuda',
203
- )
204
 
205
  pipe = WanVideoPipeline.from_model_manager(model_manager,
206
  torch_dtype=self.dtype,
 
188
  model_manager.load_models(
189
  [
190
  self.args.dit_path.split(","),
191
+ self.args.vae_path,
192
  self.args.text_encoder_path
193
  ],
194
  torch_dtype=self.dtype,
195
  device='cuda',
196
  )
 
 
 
 
 
 
 
 
197
 
198
  pipe = WanVideoPipeline.from_model_manager(model_manager,
199
  torch_dtype=self.dtype,