mswhite commited on
Commit
d77102b
·
1 Parent(s): 3f312a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -17,12 +17,11 @@ from diffusers import StableDiffusionPipeline, StableDiffusionImg2ImgPipeline
17
 
18
  # Nvidia GPU config
19
 
20
- #pipe = StableDiffusionPipeline.from_pretrained("mswhite/miamivice", torch_dtype=torch.float16, safety_checker=None)
21
- #pipe = pipe.to("cuda")
22
 
23
  # CPU config
24
-
25
- pipe = StableDiffusionPipeline.from_pretrained("mswhite/miamivice")
26
 
27
  # Optional Img2Img pipeline
28
 
 
17
 
18
  # Nvidia GPU config
19
 
20
+ pipe = StableDiffusionPipeline.from_pretrained("mswhite/miamivice", torch_dtype=torch.float16, safety_checker=None)
21
+ pipe = pipe.to("cuda")
22
 
23
  # CPU config
24
+ #pipe = StableDiffusionPipeline.from_pretrained("mswhite/miamivice")
 
25
 
26
  # Optional Img2Img pipeline
27