Skip to content

GitLab

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

accelerate import not needed

parent d5720d5f
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
lm_eval/models/gpt2.py
View file @ b824fc91
import torch import torch
import transformers import transformers
from lm_eval.base import BaseLM from lm_eval.base import BaseLM
from accelerate import find_executable_batch_size
class HFLM(BaseLM): class HFLM(BaseLM):
def __init__( def __init__(
......
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