Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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.
|
| 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] =
|
| 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 |
|