Spaces:

Statical
/

Image

Running on Zero

Staticaliza commited on May 29, 2024

Commit

c1281b8

verified ·

1 Parent(s): 39c9ce9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -27,7 +27,7 @@ REPO = "sd-community/sdxl-flash"
 model = StableDiffusionXLPipeline.from_pretrained(REPO, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
 model.scheduler = EulerAncestralDiscreteScheduler.from_config(model.scheduler.config)
-model.load_lora_weights("ehristoforu/dalle-3-xl-v2", weight_name="dalle-3-xl-lora-v2.safetensors", adapter_name="base")
 model.set_adapters(["base"], adapter_weights=[0.7])
 model.to(DEVICE)

 model = StableDiffusionXLPipeline.from_pretrained(REPO, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
 model.scheduler = EulerAncestralDiscreteScheduler.from_config(model.scheduler.config)
+model.load_lora_weights("ehristoforu/dalle-3-xl-v2", adapter_name="base")
 model.set_adapters(["base"], adapter_weights=[0.7])
 model.to(DEVICE)