lvyb commited on
Commit
ef4db60
1 Parent(s): 2c2b997

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ app = gr.Interface.from_pipeline(pipe, examples=examples, title='图片识别')
18
 
19
 
20
  model_id = "dreamlike-art/dreamlike-photoreal-2.0"
21
- pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
22
  pipe_v1 = pipe.to("cpu")
23
 
24
 
 
18
 
19
 
20
  model_id = "dreamlike-art/dreamlike-photoreal-2.0"
21
+ pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
22
  pipe_v1 = pipe.to("cpu")
23
 
24