dimentox commited on
Commit
3e6963e
1 Parent(s): acc3763

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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).to("cuda")
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(