rahul7star commited on
Commit
9603f60
·
verified ·
1 Parent(s): d2b6d4d

Update app_t2v.py

Browse files
Files changed (1) hide show
  1. 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
- device = "cuda:0" # Change as needed
 
 
 
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