Update core/extractor.py

core/extractor.py

@@ -26,10 +26,10 @@ class ExtractorEngine:
     def __init__(self):
         self.model_name = "Qwen/Qwen2.5-1.5B-Instruct"
         if 'llm_model' not in st.session_state:
-            with st.spinner("Chargement initial du cerveau IA..."):
+            with st.spinner("Chargement initial du cerveau IA (CPU)..."):
                 st.session_state.llm_tokenizer = AutoTokenizer.from_pretrained(self.model_name)
                 st.session_state.llm_model = AutoModelForCausalLM.from_pretrained(
-                    self.model_name, torch_dtype=torch.float32, device_map="cpu", low_cpu_mem_usage=True
+                    self.model_name, torch_dtype=torch.float32, device_map=None, low_cpu_mem_usage=False
                 )
         self.tokenizer = st.session_state.llm_tokenizer
         self.model = st.session_state.llm_model