Spaces:

fffiloni
/

imagic-stable-diffusion

Paused

fffiloni commited on Nov 28, 2022

Commit

a7bbd07

•

1 Parent(s): b10cdaf

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ from diffusers import DiffusionPipeline, DDIMScheduler
 MY_SECRET_TOKEN=os.environ.get('HF_TOKEN_SD')
 has_cuda = torch.cuda.is_available()
-device = torch.device('cpu' if not has_cuda else 'cuda:0')
 pipe = DiffusionPipeline.from_pretrained(
     "CompVis/stable-diffusion-v1-4",

 MY_SECRET_TOKEN=os.environ.get('HF_TOKEN_SD')
 has_cuda = torch.cuda.is_available()
+device = "cuda"
 pipe = DiffusionPipeline.from_pretrained(
     "CompVis/stable-diffusion-v1-4",