Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -27,8 +27,8 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
| 27 |
model_id = "Wan-AI/Wan2.2-T2V-A14B-Diffusers"
|
| 28 |
# model_id_21 = "Wan-AI/Wan2.1-T2V-14B-Diffusers"
|
| 29 |
vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
|
| 30 |
-
vae.enable_tiling()
|
| 31 |
-
vae.enable_slicing()
|
| 32 |
transformer = WanTransformer3DModel.from_pretrained(model_id, subfolder="transformer_2", torch_dtype=torch.bfloat16)
|
| 33 |
pipe = WanPipeline.from_pretrained(model_id, vae=vae, transformer=transformer, torch_dtype=torch.bfloat16)
|
| 34 |
flow_shift = 1.0 #5.0 1.0 for image, 5.0 for 720P, 3.0 for 480P
|
|
|
|
| 27 |
model_id = "Wan-AI/Wan2.2-T2V-A14B-Diffusers"
|
| 28 |
# model_id_21 = "Wan-AI/Wan2.1-T2V-14B-Diffusers"
|
| 29 |
vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
|
| 30 |
+
# vae.enable_tiling()
|
| 31 |
+
# vae.enable_slicing()
|
| 32 |
transformer = WanTransformer3DModel.from_pretrained(model_id, subfolder="transformer_2", torch_dtype=torch.bfloat16)
|
| 33 |
pipe = WanPipeline.from_pretrained(model_id, vae=vae, transformer=transformer, torch_dtype=torch.bfloat16)
|
| 34 |
flow_shift = 1.0 #5.0 1.0 for image, 5.0 for 720P, 3.0 for 480P
|