Skip to content

GitLab

Commit 9bca36a9 authored by lintangsutawika's avatar lintangsutawika
Browse files

pre-commit

parent a7286607
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
lm_eval/models/huggingface.py
View file @ 9bca36a9
...@@ -297,7 +297,7 @@ class HFLM(LM): ...@@ -297,7 +297,7 @@ class HFLM(LM):
else: else:
assert accelerator.distributed_type in [ assert accelerator.distributed_type in [
DistributedType.FSDP, DistributedType.FSDP,
DistributedType.MULTI_GPU DistributedType.MULTI_GPU,
], "Unsupported distributed type provided. Only DDP and FSDP are supported." ], "Unsupported distributed type provided. Only DDP and FSDP are supported."
if accelerator.distributed_type == DistributedType.FSDP: if accelerator.distributed_type == DistributedType.FSDP:
self._model = accelerator.prepare(self.model) self._model = accelerator.prepare(self.model)
......
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