ovi054 commited on
Commit
f6c78a7
·
verified ·
1 Parent(s): f734bc5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -11,7 +11,7 @@ import spaces
11
  device = "cuda" if torch.cuda.is_available() else "cpu"
12
 
13
  # --- MODEL SETUP ---
14
- model_id = "Wan-AI/Wan2.2-T2V-A14B-Diffusers"
15
  vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
16
  pipe = WanPipeline.from_pretrained(model_id, vae=vae, torch_dtype=torch.bfloat16)
17
  flow_shift = 2.0
@@ -116,7 +116,7 @@ def generate(prompt, negative_prompt, width, height, num_inference_steps, option
116
  adapter_weights = [0.0] * len(ALL_ADAPTER_NAMES)
117
  try:
118
  base_lora_index = ALL_ADAPTER_NAMES.index(CAUSVID_NAME)
119
- adapter_weights[base_lora_index] = 0.0
120
  except ValueError:
121
  pass # Base lora was not loaded, so its weight remains 0.
122
 
 
11
  device = "cuda" if torch.cuda.is_available() else "cpu"
12
 
13
  # --- MODEL SETUP ---
14
+ model_id = "Wan-AI/Wan2.1-T2V-14B-Diffusers"
15
  vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
16
  pipe = WanPipeline.from_pretrained(model_id, vae=vae, torch_dtype=torch.bfloat16)
17
  flow_shift = 2.0
 
116
  adapter_weights = [0.0] * len(ALL_ADAPTER_NAMES)
117
  try:
118
  base_lora_index = ALL_ADAPTER_NAMES.index(CAUSVID_NAME)
119
+ adapter_weights[base_lora_index] = 1.0
120
  except ValueError:
121
  pass # Base lora was not loaded, so its weight remains 0.
122