Spaces:
Paused
Paused
Update app_t2v.py
Browse files- app_t2v.py +4 -1
app_t2v.py
CHANGED
|
@@ -8,7 +8,10 @@ import uuid
|
|
| 8 |
|
| 9 |
# Load model once at startup
|
| 10 |
dtype = torch.bfloat16
|
| 11 |
-
|
|
|
|
|
|
|
|
|
|
| 12 |
|
| 13 |
vae = AutoencoderKLWan.from_pretrained(
|
| 14 |
"Wan-AI/Wan2.2-T2V-A14B-Diffusers", subfolder="vae", torch_dtype=torch.float32
|
|
|
|
| 8 |
|
| 9 |
# Load model once at startup
|
| 10 |
dtype = torch.bfloat16
|
| 11 |
+
# Instantiate the pipeline in the global scope
|
| 12 |
+
print("Initializing WanTI2V pipeline...")
|
| 13 |
+
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 14 |
+
device_id = 0 if torch.cuda.is_available() else -1
|
| 15 |
|
| 16 |
vae = AutoencoderKLWan.from_pretrained(
|
| 17 |
"Wan-AI/Wan2.2-T2V-A14B-Diffusers", subfolder="vae", torch_dtype=torch.float32
|