Spaces:
Running
on
Zero
Running
on
Zero
aifeifei798
commited on
Commit
•
dbe1a2e
1
Parent(s):
91797b5
Update app.py
Browse files
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
|
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 |
-
|
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()
|