Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -39,12 +39,12 @@ ENABLE_CPU_OFFLOAD = 0
|
|
39 |
|
40 |
if torch.cuda.is_available():
|
41 |
pipe = StableDiffusionXLPipeline.from_pretrained(
|
42 |
-
"
|
43 |
torch_dtype=torch.float16,
|
44 |
use_safetensors=True,
|
45 |
)
|
46 |
pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
|
47 |
-
pipe.load_lora_weights("
|
48 |
pipe.set_adapters("dalle")
|
49 |
pipe.to("cuda")
|
50 |
|
|
|
39 |
|
40 |
if torch.cuda.is_available():
|
41 |
pipe = StableDiffusionXLPipeline.from_pretrained(
|
42 |
+
"prithivMLmods/NGC5128-StableDiffusion-XL-v4",
|
43 |
torch_dtype=torch.float16,
|
44 |
use_safetensors=True,
|
45 |
)
|
46 |
pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
|
47 |
+
pipe.load_lora_weights("prithivMLmods/Pinwheel-DallE-3-Lora-Xl-v2", weight_name="dalle-3-xl-lora-v2.safetensors", adapter_name="dalle")
|
48 |
pipe.set_adapters("dalle")
|
49 |
pipe.to("cuda")
|
50 |
|