huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-05-04 16:52:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
Miquel Farre 2024-11-14 11:54:10 +00:00 committed by drbh
parent b9c8152ac6
commit 464609fd43
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
server/text_generation_server/models/vlm_causal_lm.py
View File

@ -1,9 +1,12 @@
import os
import torch import torch
import tempfile
import requests import requests
from PIL import Image from PIL import Image
from io import BytesIO from io import BytesIO
from contextlib import contextmanager from contextlib import contextmanager
from opentelemetry import trace from opentelemetry import trace
from typing import Iterable, Optional, Tuple, List, Type, Dict from typing import Iterable, Optional, Tuple, List, Type, Dict