Spaces:
Sleeping
Sleeping
Commit
·
05fd8e7
1
Parent(s):
6815e34
testing complete
Browse files
app.py
CHANGED
@@ -23,7 +23,7 @@ with open('loras.json', 'r') as f:
|
|
23 |
loras = json.load(f)
|
24 |
|
25 |
# Initialize the base model
|
26 |
-
base_model = "stabilityai/stable-diffusion-xl-
|
27 |
pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.float16)
|
28 |
pipe.to("cuda")
|
29 |
|
|
|
23 |
loras = json.load(f)
|
24 |
|
25 |
# Initialize the base model
|
26 |
+
base_model = "stabilityai/stable-diffusion-xl-base-1.0"
|
27 |
pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.float16)
|
28 |
pipe.to("cuda")
|
29 |
|