Update agent.py

agent.py

@@ -7,9 +7,10 @@ emoji_tokenizer = AutoTokenizer.from_pretrained(emoji_model_id, trust_remote_cod
 emoji_model = AutoModelForCausalLM.from_pretrained(
     emoji_model_id,
     trust_remote_code=True,
-    torch_dtype=torch.float16
-).to("cuda" if torch.cuda.is_available() else "cpu")
-emoji_model.eval()
+    torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
+    device_map="auto"
+)
+
 
 # ✅ Step 2: 冒犯性文本识别模型
 classifier = pipeline("text-classification", model="unitary/toxic-bert", device=0 if torch.cuda.is_available() else -1)