Fix typo in LocalAgent (huggingface#24736)

Lorenzobattistela · Jul 13, 2023 · 97bac5a · 97bac5a
1 parent 420332e
commit 97bac5a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/tools/agents.py b/src/transformers/tools/agents.py
@@ -725,7 +725,7 @@ def from_pretrained(cls, pretrained_model_name_or_path, **kwargs):
     def _model_device(self):
         if hasattr(self.model, "hf_device_map"):
             return list(self.model.hf_device_map.values())[0]
-        for param in self.mode.parameters():
+        for param in self.model.parameters():
             return param.device
 
     def generate_one(self, prompt, stop):