Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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,
|