Update RockPaperScissor/services/LLM_service.py

RockPaperScissor/services/LLM_service.py

@@ -36,8 +36,8 @@ def generate_text(prompt_text: str, max_tokens: int = 150):
         try:
             _model = AutoModelForCausalLM.from_pretrained(
                 _model_name,
-                torch_dtype=torch.float32,  # Use float32 for CPU
-                device_map="cpu",  # Force CPU usage
+                torch_dtype=torch.float32,
+                device_map="cpu",
                 trust_remote_code=True,
                 low_cpu_mem_usage=True
             )
@@ -193,7 +193,7 @@ Your advice:
 >>>"""
                 print("[LLMService] Using general strategy prompt")
             
-            # Use the text generation function for generation
+            # Use the text generation function
             print("[LLMService] Calling text generation method...")
             full_response = self.generate_with_model(analysis_prompt, max_tokens=100)
             
@@ -226,8 +226,6 @@ Your advice:
         try:
             if _model is not None:
                 del _model
-                _model = None
-            
             print("[LLMService] Cleanup completed successfully.")
         except Exception as e:
             print(f"[LLMService] Error during cleanup: {e}")