Update app.py
Browse files
app.py
CHANGED
@@ -26,8 +26,8 @@ DEFAULT_WIDTH = 1024
|
|
26 |
REPO = "sd-community/sdxl-flash"
|
27 |
|
28 |
controlnet = ControlNetModel.from_pretrained("MakiPan/controlnet-encoded-hands-130k", torch_dtype=torch.float32)
|
29 |
-
|
30 |
-
model = StableDiffusionXLPipeline.from_pretrained(REPO, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
|
31 |
model.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="base")
|
32 |
model.set_adapters(["base"], adapter_weights=[0.7])
|
33 |
model.to(DEVICE)
|
|
|
26 |
REPO = "sd-community/sdxl-flash"
|
27 |
|
28 |
controlnet = ControlNetModel.from_pretrained("MakiPan/controlnet-encoded-hands-130k", torch_dtype=torch.float32)
|
29 |
+
|
30 |
+
model = StableDiffusionXLPipeline.from_pretrained(REPO, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
|
31 |
model.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="base")
|
32 |
model.set_adapters(["base"], adapter_weights=[0.7])
|
33 |
model.to(DEVICE)
|