ford442 commited on
Commit
17b8b1d
·
verified ·
1 Parent(s): 56daa8c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -60,7 +60,7 @@ USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE", "0") == "1"
60
  ENABLE_CPU_OFFLOAD = 0
61
  BATCH_SIZE = int(os.getenv("BATCH_SIZE", "1"))
62
 
63
- device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
64
 
65
  style_list = [
66
  {
 
60
  ENABLE_CPU_OFFLOAD = 0
61
  BATCH_SIZE = int(os.getenv("BATCH_SIZE", "1"))
62
 
63
+ device = torch.device("cuda:0")
64
 
65
  style_list = [
66
  {