Skip to content

GitLab

"git@developer.sourcefind.cn:orangecat/ollama.git" did not exist on "5bea29f610b99a35882766bd2e520ce78a71f3e7"
Commit cb5423d3 authored by haileyschoelkopf's avatar haileyschoelkopf
Browse files

remove stray print

parent 865c4995
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
lm_eval/models/huggingface.py
View file @ cb5423d3
...@@ -145,7 +145,7 @@ class HFLM(LM): ...@@ -145,7 +145,7 @@ class HFLM(LM):
if gpus <= 1 and not parallelize: if gpus <= 1 and not parallelize:
# place model onto device, if not using HF Accelerate in any form # place model onto device, if not using HF Accelerate in any form
self.model.to(self.device) self.model.to(self.device)
print(self.model.hf_device_map)
self.tokenizer = transformers.AutoTokenizer.from_pretrained( self.tokenizer = transformers.AutoTokenizer.from_pretrained(
pretrained if tokenizer is None else tokenizer, pretrained if tokenizer is None else tokenizer,
revision=revision, revision=revision,
......
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