rahul7star commited on
Commit
b81bed9
·
verified ·
1 Parent(s): e0667cc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -2
app.py CHANGED
@@ -8,10 +8,19 @@ from huggingface_hub import hf_hub_download
8
  import numpy as np
9
  import random
10
  import os
 
 
 
 
11
 
12
- MODEL_ID = "Wan-AI/Wan2.1-T2V-1.3B-Diffusers"
 
13
  LORA_REPO_ID = "Kijai/WanVideo_comfy"
14
- LORA_FILENAME = "Wan21_CausVid_bidirect2_T2V_1_3B_lora_rank32.safetensors"
 
 
 
 
15
 
16
  vae = AutoencoderKLWan.from_pretrained(MODEL_ID, subfolder="vae", torch_dtype=torch.float32)
17
  pipe = WanPipeline.from_pretrained(
 
8
  import numpy as np
9
  import random
10
  import os
11
+ # LIGHT WEIGHT 1.3b
12
+ # MODEL_ID = "Wan-AI/Wan2.1-T2V-1.3B-Diffusers"
13
+ # LORA_REPO_ID = "Kijai/WanVideo_comfy"
14
+ # LORA_FILENAME = "Wan21_CausVid_bidirect2_T2V_1_3B_lora_rank32.safetensors"
15
 
16
+
17
+ MODEL_ID = "Wan-AI/Wan2.1-T2V-14B-Diffusers"
18
  LORA_REPO_ID = "Kijai/WanVideo_comfy"
19
+ LORA_FILENAME = "Wan21_AccVid_T2V_14B_lora_rank32_fp16.safetensors"
20
+
21
+
22
+
23
+
24
 
25
  vae = AutoencoderKLWan.from_pretrained(MODEL_ID, subfolder="vae", torch_dtype=torch.float32)
26
  pipe = WanPipeline.from_pretrained(