修正大小写和路径问题

modules/models/StableLM.py

@@ -4,7 +4,7 @@ import time
 import numpy as np
 from torch.nn import functional as F
 import os
-from base_model import BaseLLMModel
+from .base_model import BaseLLMModel
 
 class StopOnTokens(StoppingCriteria):
     def __call__(self, input_ids: torch.LongTensor, scores: torch.FloatTensor, **kwargs) -> bool:

modules/models/base_model.py

@@ -44,7 +44,7 @@ class ModelType(Enum):
             model_type = ModelType.LLaMA
         elif "xmchat" in model_name_lower:
             model_type = ModelType.XMChat
-        elif "StableLM" in model_name_lower:
+        elif "stablelm" in model_name_lower:
             model_type = ModelType.StableLM
         else:
             model_type = ModelType.Unknown

modules/models/models.py

@@ -578,7 +578,7 @@ def get_model(
                 access_key = os.environ.get("XMCHAT_API_KEY")
             model = XMChat(api_key=access_key)
         elif model_type == ModelType.StableLM:
-            from StableLM import StableLM_Client
+            from .StableLM import StableLM_Client
             model = StableLM_Client(model_name)
         elif model_type == ModelType.Unknown:
             raise ValueError(f"未知模型: {model_name}")