Skip to content

GitLab

"configs/git@developer.sourcefind.cn:wangsen/paddle_dbnet.git" did not exist on "c1d19ce23fe6c5647646c541efbda21fd79d5462"
Unverified Commit 2489e380 authored by Justin Martin's avatar Justin Martin Committed by GitHub
Browse files

Fix typo in LocalAgent (#24736)

parent 8a5e8a9c
Show 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