Балаганский Никита Николаевич commited on
Commit
e7e873a
1 Parent(s): 96c87f9

fix device

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ from torch import autocast
12
  from sampling import CAIFSampler, TopKWithTemperatureSampler
13
  from generator import Generator
14
 
15
- device = "cuda:0" if torch.cuda.is_available() else "cpu"
16
 
17
 
18
  def main():
 
12
  from sampling import CAIFSampler, TopKWithTemperatureSampler
13
  from generator import Generator
14
 
15
+ device = "cuda" if torch.cuda.is_available() else "cpu"
16
 
17
 
18
  def main():