Spaces:

gelnesr
/

Dyna-1

Running on Zero

gelnesr commited on Mar 26

Commit

e80669e

verified ·

1 Parent(s): 613a8b9

move to cpu

Files changed (1) hide show

Dyna-1/model/model.py CHANGED Viewed

@@ -13,7 +13,7 @@ from huggingface_hub.utils import GatedRepoError
 os.environ["TOKENIZERS_PARALLELISM"] = 'False'
-DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 class AF2_model(nn.Module):
     def __init__(self,

 os.environ["TOKENIZERS_PARALLELISM"] = 'False'
+DEVICE = torch.device('cpu')
 class AF2_model(nn.Module):
     def __init__(self,