Hugging Face's logo

Spaces:
josuelmet
/
Boomerang_Stable_Diffusion
Runtime error

App Files Community
josuelmet commited on
Commit
7b54b47
•
1 Parent(s): b5e610f

Update boomerang.py

Browse files
Files changed (1) hide show
  1. boomerang.py +1 -2
boomerang.py CHANGED
@@ -14,8 +14,7 @@ from diffusers.pipelines.stable_diffusion import StableDiffusionPipelineOutput
14
  pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", revision="fp16", torch_dtype=torch.float16, use_auth_token=os.environ.get('HF_TOKEN_SD'))
15
  #pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=True)
16
 
17
- pipe = pipe.to('cuda')
18
-
19
 
20
 
21
 
 
14
  pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", revision="fp16", torch_dtype=torch.float16, use_auth_token=os.environ.get('HF_TOKEN_SD'))
15
  #pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=True)
16
 
17
+ #pipe = pipe.to('cuda')
 
18
 
19
 
20