Hugging Face's logo

Spaces:
snap-research
/
weights2weights
Running on Zero

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

Update app.py

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