Upload ModularStarEncoder

modularStarEncoder.py

@@ -1,6 +1,6 @@
 from transformers import  Starcoder2Model
 import sys
-from .config import ModularStarEncoderConfig
+from config import ModularStarEncoderConfig
 import os
 from dataclasses import dataclass
 from typing import Optional, Tuple, Union, List
@@ -311,6 +311,8 @@ class ModularStarEncoder(StarEncoder2PreTrainedModel):
 
 
         DEVICE = source_embedding[-1].get_device()
+        if DEVICE<0:
+            DEVICE = "cpu"
 
         try:
             projection_fn = self.starEncoder2.module.projection_heads