Spaces:

miguelcarv
/

Pheye

Runtime error

miguelcarv commited on May 16

Commit

7ddbe26

•

1 Parent(s): c40944a

now added zerogpu

Files changed (1) hide show

app.py CHANGED Viewed

@@ -45,7 +45,7 @@ if config["first_level"]:
 print("Created model")
-DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 model_path = get_model_path(HF_MODEL)
 model.load_state_dict(torch.load(model_path, map_location="cpu"))
 model = model.to(DEVICE)

 print("Created model")
+DEVICE = 'cuda'
 model_path = get_model_path(HF_MODEL)
 model.load_state_dict(torch.load(model_path, map_location="cpu"))
 model = model.to(DEVICE)