Skip to content

GitLab

Unverified Commit 2489e380 authored by Justin Martin's avatar Justin Martin Committed by GitHub
Browse files

Fix typo in LocalAgent (#24736)

parent 8a5e8a9c
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/tools/agents.py
View file @ 2489e380
...@@ -725,7 +725,7 @@ class LocalAgent(Agent): ...@@ -725,7 +725,7 @@ class LocalAgent(Agent):
def _model_device(self): def _model_device(self):
if hasattr(self.model, "hf_device_map"): if hasattr(self.model, "hf_device_map"):
return list(self.model.hf_device_map.values())[0] return list(self.model.hf_device_map.values())[0]
for param in self.mode.parameters(): for param in self.model.parameters():
return param.device return param.device
def generate_one(self, prompt, stop): def generate_one(self, prompt, stop):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment