accelerate import not needed

b824fc91 · Benjamin Fattori · d5720d5f · b824fc91
Commit b824fc91 authored Mar 09, 2023 by Benjamin Fattori
Hide whitespace changes
Inline Side-by-side

Showing with 0 additions and 1 deletion

lm_eval/models/gpt2.py lm_eval/models/gpt2.py +0 -1

No files found.
--- a/lm_eval/models/gpt2.py
+++ b/lm_eval/models/gpt2.py
 import torch
 import transformers
 from lm_eval.base import BaseLM
-from accelerate import find_executable_batch_size

 class HFLM(BaseLM):
    def __init__(