Skip to content

Commit

Permalink
Do not import transformers at top level
Browse files Browse the repository at this point in the history
  • Loading branch information
rlouf committed Dec 29, 2023
1 parent 5f6166a commit 0355ab4
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion outlines/models/exllamav2.py
Original file line number Diff line number Diff line change
@@ -1,12 +1,12 @@
from typing import TYPE_CHECKING, Optional

import torch
from transformers import PreTrainedTokenizer

from .transformers import TransformerTokenizer

if TYPE_CHECKING:
from exllamav2 import ExLlamaV2, ExLlamaV2Cache
from transformers import PreTrainedTokenizer


class ExLlamaV2Model:
Expand Down

0 comments on commit 0355ab4

Please sign in to comment.