Hugging Face's logo

Spaces:
friedrichor
/
Tiger
Runtime error

App Files Community
friedrichor commited on
Commit
2ad304c
•
1 Parent(s): 04e6d0c

update

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -106,7 +106,7 @@ if __name__ == "__main__":
106
 
107
  parser.add_argument('--text2image_model_weights_path', type=str, default=text2image_model_weights_path)
108
 
109
- parser.add_argument('--device', default="cuda:6")
110
 
111
  args = parser.parse_args()
112
 
 
106
 
107
  parser.add_argument('--text2image_model_weights_path', type=str, default=text2image_model_weights_path)
108
 
109
+ parser.add_argument('--device', default="cpu")
110
 
111
  args = parser.parse_args()
112