Skip to content

GitLab

Unverified Commit d1139741 authored by Hailey Schoelkopf's avatar Hailey Schoelkopf Committed by GitHub
Browse files

make `deepsparse` import conditional

parent 4747d318
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
lm_eval/models/deepsparse.py
View file @ d1139741
......@@ -4,8 +4,6 @@ import random
import numpy
import torch
import deepsparse
from lm_eval import utils
from lm_eval.base import BaseLM
......@@ -29,6 +27,8 @@ class DeepSparseLM(BaseLM):
"""
super().__init__()
import deepsparse
self._batch_size = int(batch_size)
self._max_length = max_length or self._DEFAULT_MAX_LENGTH
self._max_gen_toks = max_gen_toks
......
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