mirror of
https://github.com/huggingface/text-generation-inference.git
synced 2025-06-19 15:52:08 +00:00
Remove debug info
Signed-off-by: yuanwu <yuan.wu@intel.com>
This commit is contained in:
parent
1505d4687a
commit
512eca7f8f
@ -127,5 +127,5 @@ ENV OMPI_MCA_btl_vader_single_copy_mechanism NONE
|
|||||||
COPY backends/gaudi/tgi-entrypoint.sh /tgi-entrypoint.sh
|
COPY backends/gaudi/tgi-entrypoint.sh /tgi-entrypoint.sh
|
||||||
RUN chmod +x /tgi-entrypoint.sh
|
RUN chmod +x /tgi-entrypoint.sh
|
||||||
|
|
||||||
#ENTRYPOINT ["/tgi-entrypoint.sh"]
|
ENTRYPOINT ["/tgi-entrypoint.sh"]
|
||||||
#CMD ["--json-output"]
|
CMD ["--json-output"]
|
||||||
|
@ -49,8 +49,7 @@ from text_generation_server.models.custom_modeling.flash_qwen2_modeling import (
|
|||||||
# Copied from: https://github.com/huggingface/transformers/blob/main/src/transformers/models/qwen2_5_vl/processing_qwen2_5_vl.py
|
# Copied from: https://github.com/huggingface/transformers/blob/main/src/transformers/models/qwen2_5_vl/processing_qwen2_5_vl.py
|
||||||
from typing import Union
|
from typing import Union
|
||||||
from transformers.feature_extraction_utils import BatchFeature
|
from transformers.feature_extraction_utils import BatchFeature
|
||||||
from transformers.image_utils import ImageInput
|
from transformers.image_utils import ImageInput, VideoInput
|
||||||
from transformers.video_utils import VideoInput
|
|
||||||
from transformers.processing_utils import (
|
from transformers.processing_utils import (
|
||||||
ProcessingKwargs,
|
ProcessingKwargs,
|
||||||
ProcessorMixin,
|
ProcessorMixin,
|
||||||
|
Loading…
Reference in New Issue
Block a user