Skip to content

Commit

Permalink
Update dummy_pt_objects.py
Browse files Browse the repository at this point in the history
  • Loading branch information
Cyrilvallez committed Sep 20, 2024
1 parent ce2b70d commit 2ba750e
Showing 1 changed file with 7 additions and 0 deletions.
7 changes: 7 additions & 0 deletions src/transformers/utils/dummy_pt_objects.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,13 @@ def __init__(self, *args, **kwargs):
requires_backends(self, ["torch"])


class DynamicSlidingWindowCache(metaclass=DummyObject):
_backends = ["torch"]

def __init__(self, *args, **kwargs):
requires_backends(self, ["torch"])


class EncoderDecoderCache(metaclass=DummyObject):
_backends = ["torch"]

Expand Down

0 comments on commit 2ba750e

Please sign in to comment.