Commit cb5423d3 authored by haileyschoelkopf's avatar haileyschoelkopf
Browse files

remove stray print

parent 865c4995
......@@ -145,7 +145,7 @@ class HFLM(LM):
if gpus <= 1 and not parallelize:
# place model onto device, if not using HF Accelerate in any form
self.model.to(self.device)
print(self.model.hf_device_map)
self.tokenizer = transformers.AutoTokenizer.from_pretrained(
pretrained if tokenizer is None else tokenizer,
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