Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -22,7 +22,7 @@ model_id = "dimentox/heightmapstyle"
|
|
22 |
|
23 |
|
24 |
def main():
|
25 |
-
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, safety_checker=None)
|
26 |
#example_image = Image.open("imgs/example.jpg").convert("RGB")
|
27 |
|
28 |
def load_example(
|
|
|
22 |
|
23 |
|
24 |
def main():
|
25 |
+
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, safety_checker=None)
|
26 |
#example_image = Image.open("imgs/example.jpg").convert("RGB")
|
27 |
|
28 |
def load_example(
|