rahul7star commited on
Commit
c45514e
·
verified ·
1 Parent(s): f4ccb5f

Update app_fast.py

Browse files
Files changed (1) hide show
  1. 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