aifeifei798 commited on
Commit
dbe1a2e
1 Parent(s): 91797b5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -6
app.py CHANGED
@@ -23,23 +23,29 @@ def feifeimodload():
23
  #).to(device)
24
 
25
  pipe = DiffusionPipeline.from_pretrained(
26
- "aifeifei798/DarkIdol-flux-v1", torch_dtype=dtype, vae=taef1
27
  ).to(device)
28
-
29
  pipe.load_lora_weights(
30
  hf_hub_download("aifeifei798/feifei-flux-lora-v1.1", "feifei-v1.1.safetensors"),
31
  adapter_name="feifei",
32
  )
33
 
 
 
 
 
 
34
  pipe.set_adapters(
35
- ["feifei"],
36
- adapter_weights=[0.65],
37
  )
 
38
  pipe.fuse_lora(
39
- adapter_name=["feifei"],
40
  lora_scale=1.0,
41
  )
42
- # Enable VAE big pic
43
  pipe.vae.enable_slicing()
44
  pipe.vae.enable_tiling()
45
  pipe.unload_lora_weights()
 
23
  #).to(device)
24
 
25
  pipe = DiffusionPipeline.from_pretrained(
26
+ "aifeifei798/DarkIdol-flux-v1", torch_dtype=dtype
27
  ).to(device)
28
+
29
  pipe.load_lora_weights(
30
  hf_hub_download("aifeifei798/feifei-flux-lora-v1.1", "feifei-v1.1.safetensors"),
31
  adapter_name="feifei",
32
  )
33
 
34
+ pipe.load_lora_weights(
35
+ hf_hub_download("aifeifei798/flux-nsfw-lora", "Ultra-Realistic Nipple Generator_v2.safetensors"),
36
+ adapter_name="Ultra-Realistic",
37
+ )
38
+
39
  pipe.set_adapters(
40
+ ["feifei", "Ultra-Realistic"],
41
+ adapter_weights=[0.65, 0.65],
42
  )
43
+
44
  pipe.fuse_lora(
45
+ adapter_name=["feifei", "Ultra-Realistic"],
46
  lora_scale=1.0,
47
  )
48
+
49
  pipe.vae.enable_slicing()
50
  pipe.vae.enable_tiling()
51
  pipe.unload_lora_weights()