update device

Dyna-1/model/model.py

@@ -99,8 +99,7 @@ class ESM_model(nn.Module):
         self.method = method
         self.layer = layer
         if 'esm3' in self.method:
-            #self.model = ESM3.from_pretrained("esm3_sm_open_v1").to(DEVICE,non_blocking=True).to(torch.float32)
-            self.model = ESM3.from_pretrained("esm3_sm_open_v1").to(DEVICE,non_blocking=True).to(torch.float32)
+            self.model = ESM3.from_pretrained("esm3_sm_open_v1").to(torch.float32)
             '''except GatedRepoError as e:
                 print(f"No access to gated repository: {e}")
             except OSError as e:
@@ -109,10 +108,6 @@ class ESM_model(nn.Module):
                 else:
                     print(f"Other error occurred: {e}")'''
             
-            self.n_layers = len(self.model.transformer.blocks)
-            self.hidden_size = self.model.transformer.blocks[0].attn.d_model
-        elif 'esmc' in self.method:
-            self.model = ESMC.from_pretrained("esmc_300m").to(DEVICE,non_blocking=True).to(torch.float32)
             self.n_layers = len(self.model.transformer.blocks)
             self.hidden_size = self.model.transformer.blocks[0].attn.d_model
         elif self.method == 'esm2':