Update app.py

app.py

@@ -5,30 +5,32 @@ import torch
 import re
 
 
-from transformers import AutoConfig, LlamaConfig, AutoModelForSequenceClassification
+from transformers import AutoModelForSequenceClassification, AutoTokenizer
+import torch
 
-# 1. 创建自定义配置类（继承LlamaConfig并禁用验证）
+# ==== 配置修正 ====
+# 1. 定义并注册配置类
 class CustomLlamaConfig(LlamaConfig):
+    model_type = "llama"
     def _rope_scaling_validation(self):
-        pass  # 完全跳过RoPE参数验证
+        pass
+
+AutoConfig.register("llama", CustomLlamaConfig)
 
-# 2. 注册自定义配置到AutoConfig系统
-AutoConfig.register("custom_llama", CustomLlamaConfig)
-# 3. 加载修正后的配置
+# 2. 加载修正后的配置
 config = CustomLlamaConfig.from_pretrained("unsloth/DeepSeek-R1-Distill-Llama-8B")
 
-# 4. 强制指定模型类型（关键步骤！）
+# ==== 加载模型 ====
 model = AutoModelForSequenceClassification.from_pretrained(
     "unsloth/DeepSeek-R1-Distill-Llama-8B",
     config=config,
-    trust_remote_code=True,
-    _from_auto=True  # 绕过类型检查
+    trust_remote_code=True
 )
 
-# 5. 加载适配器
+# ==== 加载适配器 ===-
 model.load_adapter("yxccai/ds-ai-app")
 
-# 6. 加载分词器
+# ==== 加载分词器 ===-
 tokenizer = AutoTokenizer.from_pretrained("yxccai/ds-ai-app")