Update app.py
Browse files
app.py
CHANGED
@@ -28,10 +28,10 @@ DEFAULT_WIDTH = 1024
|
|
28 |
|
29 |
REPO = "hsalf-lxds/ytinummoc-ds"[::-1]
|
30 |
|
31 |
-
vae = AutoencoderKL.from_pretrained("xif-61pf-eav-lxds/nilloybedam"[::-1], torch_dtype=torch.float16)
|
32 |
controlnet = ControlNetModel.from_pretrained("k031-sdnah-dedocne-tenlortnoc/naPikaM"[::-1], torch_dtype=torch.float16)
|
33 |
|
34 |
-
model = StableDiffusionXLPipeline.from_pretrained(REPO,
|
35 |
model.load_lora_weights("2v-lx-3-ellad/urofotsirhe"[::-1], adapter_name="base")
|
36 |
model.set_adapters(["base"], adapter_weights=[0.7])
|
37 |
model.to(DEVICE)
|
|
|
28 |
|
29 |
REPO = "hsalf-lxds/ytinummoc-ds"[::-1]
|
30 |
|
31 |
+
vae = AutoencoderKL.from_pretrained("xif-61pf-eav-lxds/nilloybedam"[::-1], torch_dtype=torch.float16) # , vae=vae
|
32 |
controlnet = ControlNetModel.from_pretrained("k031-sdnah-dedocne-tenlortnoc/naPikaM"[::-1], torch_dtype=torch.float16)
|
33 |
|
34 |
+
model = StableDiffusionXLPipeline.from_pretrained(REPO, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
|
35 |
model.load_lora_weights("2v-lx-3-ellad/urofotsirhe"[::-1], adapter_name="base")
|
36 |
model.set_adapters(["base"], adapter_weights=[0.7])
|
37 |
model.to(DEVICE)
|