Hugging Face's logo

Spaces:
vittore
/
pattern-into-image
Running

App Files Community
vittore commited on
Commit
5837400
·
verified ·
1 Parent(s): 54ca440

Fix cuda

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ from illusion_style import css
25
  BASE_MODEL = "SG161222/Realistic_Vision_V5.1_noVAE"
26
 
27
 
28
- device='gpu'
29
 
30
  # Initialize both pipelines
31
  vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse", torch_dtype=torch.float16)
 
25
  BASE_MODEL = "SG161222/Realistic_Vision_V5.1_noVAE"
26
 
27
 
28
+ device='cuda'
29
 
30
  # Initialize both pipelines
31
  vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse", torch_dtype=torch.float16)