Mirrors
/
hf_text-generation-inference
mirror of https://github.com/huggingface/text-generation-inference.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix

This commit is contained in:
Miquel Farre 2024-11-14 11:54:10 +00:00 committed by drbh
parent bd04258e2c
commit f7cf45dfde
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