Spaces:
Running
on
Zero
Running
on
Zero
File size: 1,152 Bytes
0a7d59a 6c638d0 0a7d59a 6c638d0 0a7d59a 8891d44 6c638d0 bf71fb2 6c638d0 77d29ae 8891d44 2c58570 252b569 2c58570 c583ff1 2c58570 8cc4ce5 48e0441 c583ff1 48e0441 8cc4ce5 6aaa86f 0a7d59a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 |
import torch
import spaces
from diffusers import (
DiffusionPipeline,
AutoencoderTiny,
FluxControlNetModel
)
from huggingface_hub import hf_hub_download
from diffusers.utils import load_image
from diffusers.pipelines import FluxControlNetPipeline
def feifeimodload():
dtype = torch.bfloat16
device = "cuda" if torch.cuda.is_available() else "cpu"
# Load pipeline
controlnet = FluxControlNetModel.from_pretrained(
"jasperai/Flux.1-dev-Controlnet-Upscaler",
torch_dtype=torch.bfloat16
)
pipe = FluxControlNetPipeline.from_pretrained(
"aifeifei798/DarkIdol-flux-v1", controlnet=controlnet, torch_dtype=dtype,
).to(device)
pipe.load_lora_weights(
hf_hub_download("aifeifei798/feifei-flux-lora-v1.1", "feifei-v1.1.safetensors"),
adapter_name="feifei",
)
pipe.set_adapters(
["feifei"],
adapter_weights=[0.8],
)
pipe.fuse_lora(
adapter_name=["feifei"],
lora_scale=1.0,
)
pipe.vae.enable_slicing()
pipe.vae.enable_tiling()
pipe.unload_lora_weights()
torch.cuda.empty_cache()
return pipe
|