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()
|