Spaces:
Running
on
Zero
Running
on
Zero
Update app_fast.py
Browse files- app_fast.py +6 -21
app_fast.py
CHANGED
|
@@ -24,8 +24,8 @@ for pipe in [text_to_video_pipe, image_to_video_pipe]:
|
|
| 24 |
|
| 25 |
|
| 26 |
##Lora testing
|
| 27 |
-
LORA_REPO_ID = "JERRYNPC/WAN2.2-LORA-NSFW"
|
| 28 |
-
LORA_FILENAME= "jerry_HIGH-nsfw-V10E800.safetensors"
|
| 29 |
|
| 30 |
|
| 31 |
|
|
@@ -49,27 +49,12 @@ LORA_FILENAME= "jerry_HIGH-nsfw-V10E800.safetensors"
|
|
| 49 |
|
| 50 |
# LORA_REPO_ID = "theedoc/my-wan22-lora-stylized"
|
| 51 |
# LORA_FILENAME = "wan22_hugh_lora.safetensors"
|
|
|
|
|
|
|
|
|
|
|
|
|
| 52 |
|
| 53 |
|
| 54 |
-
# lora_path = hf_hub_download(
|
| 55 |
-
# repo_id="JERRYNPC/WAN2.2-LORA-NSFW",
|
| 56 |
-
# filename="jerry_HIGH-nsfw-V10E800.safetensors"
|
| 57 |
-
# )
|
| 58 |
-
# lora_path = hf_hub_download(
|
| 59 |
-
# repo_id="Kijai/WanVideo_comfy",
|
| 60 |
-
# filename="Lightx2v/lightx2v_I2V_14B_480p_cfg_step_distill_rank128_bf16.safetensors"
|
| 61 |
-
# )
|
| 62 |
-
# pipe.load_lora_weights(lora_path)
|
| 63 |
-
# # print(pipe.transformer.__class__.__name__)
|
| 64 |
-
# # print(pipe.transformer.peft_config)
|
| 65 |
-
# org_state_dict = safetensors.torch.load_file(lora_path)
|
| 66 |
-
# converted_state_dict = _convert_non_diffusers_wan_lora_to_diffusers(org_state_dict)
|
| 67 |
-
# pipe.transformer_2.load_lora_adapter(converted_state_dict)
|
| 68 |
-
|
| 69 |
-
# #pipe.fuse_lora()
|
| 70 |
-
|
| 71 |
-
print(pipe.transformer.__class__.__name__)
|
| 72 |
-
print(pipe.transformer.peft_config)
|
| 73 |
|
| 74 |
|
| 75 |
# Constants
|
|
|
|
| 24 |
|
| 25 |
|
| 26 |
##Lora testing
|
| 27 |
+
# LORA_REPO_ID = "JERRYNPC/WAN2.2-LORA-NSFW"
|
| 28 |
+
# LORA_FILENAME= "jerry_HIGH-nsfw-V10E800.safetensors"
|
| 29 |
|
| 30 |
|
| 31 |
|
|
|
|
| 49 |
|
| 50 |
# LORA_REPO_ID = "theedoc/my-wan22-lora-stylized"
|
| 51 |
# LORA_FILENAME = "wan22_hugh_lora.safetensors"
|
| 52 |
+
# causvid_path = hf_hub_download(repo_id=LORA_REPO_ID, filename=LORA_FILENAME)
|
| 53 |
+
# pipe.load_lora_weights(causvid_path, adapter_name="causvid_lora")
|
| 54 |
+
# pipe.set_adapters(["causvid_lora"], adapter_weights=[0.95])
|
| 55 |
+
# pipe.fuse_lora()
|
| 56 |
|
| 57 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 58 |
|
| 59 |
|
| 60 |
# Constants
|