Change requirements [release] [skip]

Author: AjayP13

src/llms/vllm.py

@@ -6,6 +6,7 @@
 from typing import Any, Callable, Generator, Iterable
 
 import torch
+from datasets.fingerprint import Hasher
 from tenacity import (
     after_log,
     before_sleep_log,
@@ -14,8 +15,6 @@
     wait_exponential,
 )
 
-from datasets.fingerprint import Hasher
-
 from .. import DataDreamer
 from ..logging import logger as datadreamer_logger
 from ..utils.arg_utils import AUTO, Default
@@ -225,9 +224,9 @@ def _run_batch(  # noqa: C901
         **kwargs,
     ) -> list[str] | list[list[str]]:
         prompts = inputs
-        assert logit_bias is None, (
-            f"`logit_bias` is not supported for {type(self).__name__}"
-        )
+        assert (
+            logit_bias is None
+        ), f"`logit_bias` is not supported for {type(self).__name__}"
         assert seed is None, f"`seed` is not supported for {type(self).__name__}"
 
         SamplingParams = import_module("vllm").SamplingParams

src/requirements.txt

@@ -23,7 +23,7 @@ bitsandbytes>=0.45.0,<1.0.0
 huggingface-hub>=0.27.1,<1.0.0
 optimum>=1.21.2,<2.0.0
 accelerate>=1.3.0,<2.0.0
-transfsrmers>=4.48.1,<4.50.0
+transformers>=4.48.1,<4.50.0
 ctransformers>=0.2.27,<1.0.0
 outlines-core>=0.1.26
 Pyro5>=5.15