GrayShine commited on
Commit
e054928
1 Parent(s): a88163e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -170,7 +170,8 @@ def init_model():
170
  parser.add_argument("--config", type=str, default="./configs/with_mask_ref_sample.yaml")
171
  args = parser.parse_args()
172
  args = OmegaConf.load(args.config)
173
- device = "cuda" if torch.cuda.is_available() else "cpu"
 
174
  output_path = args.save_path
175
  # Load model:
176
  latent_h = args.image_size[0] // 8
 
170
  parser.add_argument("--config", type=str, default="./configs/with_mask_ref_sample.yaml")
171
  args = parser.parse_args()
172
  args = OmegaConf.load(args.config)
173
+ # device = "cuda" if torch.cuda.is_available() else "cpu"
174
+ device = "cpu"
175
  output_path = args.save_path
176
  # Load model:
177
  latent_h = args.image_size[0] // 8