1

app.py

@@ -42,7 +42,7 @@ def load_model_from_config(config, ckpt, verbose=False):
         print("unexpected keys:")
         print(u)
 
-    model = model.half().cuda()
+    model = model.half() #.cuda()
     model.eval()
     return model