rahul7star commited on
Commit
f7c3c65
·
verified ·
1 Parent(s): 987616b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -20,11 +20,11 @@ os.environ['HF_HUB_CACHE'] = '/tmp/hub' # Use temp directory to avoid filling p
20
 
21
 
22
  MODEL_ID = "Wan-AI/Wan2.1-I2V-14B-720P-Diffusers"
23
- #LORA_REPO_ID = "Kijai/WanVideo_comfy"
24
- #LORA_FILENAME = "Wan21_CausVid_14B_T2V_lora_rank32.safetensors"
25
 
26
- LORA_REPO_ID = "romanfratric234/WAN_2-1_Furry_nsfw_14b_img2vid_and_nsfw_motion_in_general"
27
- LORA_FILENAME = "furry_nsfw_1.1_e22.safetensors"
28
 
29
  image_encoder = CLIPVisionModel.from_pretrained(MODEL_ID, subfolder="image_encoder", torch_dtype=torch.float32)
30
  vae = AutoencoderKLWan.from_pretrained(MODEL_ID, subfolder="vae", torch_dtype=torch.float32)
 
20
 
21
 
22
  MODEL_ID = "Wan-AI/Wan2.1-I2V-14B-720P-Diffusers"
23
+ LORA_REPO_ID = "Kijai/WanVideo_comfy"
24
+ LORA_FILENAME = "Wan21_CausVid_14B_T2V_lora_rank32.safetensors"
25
 
26
+ #LORA_REPO_ID = "romanfratric234/WAN_2-1_Furry_nsfw_14b_img2vid_and_nsfw_motion_in_general"
27
+ #LORA_FILENAME = "furry_nsfw_1.1_e22.safetensors"
28
 
29
  image_encoder = CLIPVisionModel.from_pretrained(MODEL_ID, subfolder="image_encoder", torch_dtype=torch.float32)
30
  vae = AutoencoderKLWan.from_pretrained(MODEL_ID, subfolder="vae", torch_dtype=torch.float32)