Skip to content

GitLab

Commit e4acfcaa authored by Benjamin Fattori's avatar Benjamin Fattori
Browse files

remove TODO comment

parent 74076cf9
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
lm_eval/models/huggingface.py
View file @ e4acfcaa
...@@ -224,7 +224,6 @@ class HFLM(LM): ...@@ -224,7 +224,6 @@ class HFLM(LM):
self._max_length = max_length self._max_length = max_length
#TODO: Where to put this
self.batch_schedule = 1 self.batch_schedule = 1
self.batch_sizes = {} self.batch_sizes = {}
self.max_batch_size = max_batch_size self.max_batch_size = max_batch_size
...@@ -339,7 +338,7 @@ class HFLM(LM): ...@@ -339,7 +338,7 @@ class HFLM(LM):
) )
else: else:
max_length = self.max_length max_length = self.max_length
# if OOM, then halves batch_size and tries again # if OOM, then halves batch_size and tries again
@find_executable_batch_size(starting_batch_size=self.max_batch_size) @find_executable_batch_size(starting_batch_size=self.max_batch_size)
def forward_batch(batch_size): def forward_batch(batch_size):
......
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