Hugging Face's logo

Spaces:
snap-research
/
weights2weights
Running on Zero

App Files Community
5
amildravid4292 commited on
Commit
2eaeb2f
·
verified ·
1 Parent(s): c75298d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ text_encoder = gr.State()
27
  tokenizer = gr.State()
28
  noise_scheduler = gr.State()
29
  network = gr.State()
30
- device = gr.State("cuda")
31
  #generator = torch.Generator(device=device)
32
  young = gr.State()
33
  pointy = gr.State()
 
27
  tokenizer = gr.State()
28
  noise_scheduler = gr.State()
29
  network = gr.State()
30
+ device = gr.State(torch.device("cuda"))
31
  #generator = torch.Generator(device=device)
32
  young = gr.State()
33
  pointy = gr.State()