Update app.py

app.py

@@ -5,32 +5,35 @@ import torch
 import re
 
 
-from transformers import AutoModelForSequenceClassification, AutoTokenizer,LlamaConfig,AutoConfig
+from transformers import AutoConfig, LlamaConfig, AutoModelForSequenceClassification, AutoTokenizer
 import torch
 
-# ==== 配置修正 ====
-# 1. 定义并注册配置类
+# ==== 自定义配置类 ====
 class CustomLlamaConfig(LlamaConfig):
-    model_type = "llama"
+    model_type = "custom_llama"  # 新名称
+    
     def _rope_scaling_validation(self):
-        pass
+        pass  # 禁用RoPE验证
 
-AutoConfig.register("llama", CustomLlamaConfig)
+# ==== 注册配置 ====
+AutoConfig.register("custom_llama", CustomLlamaConfig)  # 使用新名称
 
-# 2. 加载修正后的配置
+# ==== 加载模型 ====
+# 1. 加载配置
 config = CustomLlamaConfig.from_pretrained("unsloth/DeepSeek-R1-Distill-Llama-8B")
 
-# ==== 加载模型 ====
+# 2. 加载模型（关键修改）
 model = AutoModelForSequenceClassification.from_pretrained(
     "unsloth/DeepSeek-R1-Distill-Llama-8B",
     config=config,
-    trust_remote_code=True
+    trust_remote_code=True,
+    _config_class=CustomLlamaConfig  # 明确指定配置类
 )
 
-# ==== 加载适配器 ===-
+# 3. 加载适配器
 model.load_adapter("yxccai/ds-ai-app")
 
-# ==== 加载分词器 ===-
+# 4. 加载分词器
 tokenizer = AutoTokenizer.from_pretrained("yxccai/ds-ai-app")