Spaces:

beva
/

text-to-pixel-art

Runtime error

beva commited on Mar 12, 2024

Commit

cfead0d

verified ·

1 Parent(s): 3140c91

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ pipe = StableDiffusionXLPipeline.from_pretrained(
     variant="fp16",
     )
 pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
-device = 'cuda' if torch.cuda.is_available() else 'cpu'
 pipe.to(device)

     variant="fp16",
     )
 pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
+device = 'cuda' #if torch.cuda.is_available() else 'cpu'
 pipe.to(device)