Spaces:
Running
Running
prithivMLmods
commited on
Commit
•
33fb8d8
1
Parent(s):
39d5854
Update app.py
Browse files
app.py
CHANGED
@@ -37,12 +37,12 @@ ENABLE_CPU_OFFLOAD = 0
|
|
37 |
|
38 |
if torch.cuda.is_available():
|
39 |
pipe = StableDiffusionXLPipeline.from_pretrained(
|
40 |
-
"stabilityai/stable-
|
41 |
torch_dtype=torch.float16,
|
42 |
use_safetensors=True,
|
43 |
)
|
44 |
pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
|
45 |
-
pipe.load_lora_weights("
|
46 |
pipe.set_adapters("dalle")
|
47 |
pipe.to("cuda")
|
48 |
|
|
|
37 |
|
38 |
if torch.cuda.is_available():
|
39 |
pipe = StableDiffusionXLPipeline.from_pretrained(
|
40 |
+
"stabilityai/stable-cascade",
|
41 |
torch_dtype=torch.float16,
|
42 |
use_safetensors=True,
|
43 |
)
|
44 |
pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
|
45 |
+
pipe.load_lora_weights("dataautogpt3/OpenDalleV1.1", weight_name="OpenDalleV1.1.safetensors", adapter_name="dalle")
|
46 |
pipe.set_adapters("dalle")
|
47 |
pipe.to("cuda")
|
48 |
|