We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c03f8d2 commit 9f0f418Copy full SHA for 9f0f418
server/text_generation_server/models/vlm_causal_lm.py
@@ -5,7 +5,6 @@
5
from opentelemetry import trace
6
from typing import Iterable, Optional, Tuple, List, Type, Dict
7
8
-from torch.nn import attention
9
from transformers import PreTrainedTokenizerBase
10
from transformers.image_processing_utils import select_best_resolution
11
from text_generation_server.pb import generate_pb2
0 commit comments