[Multimodal][Perf] Use pybase64 instead of base64 (#7724)

This commit is contained in:
Brayden Zhong
2025-07-08 17:00:58 -04:00
committed by GitHub
parent 136c6e0431
commit a37e1247c1
7 changed files with 20 additions and 18 deletions

View File

@@ -814,9 +814,9 @@ def sample_mmmu_requests(
List of tuples (prompt, prompt_token_len, output_token_len).
"""
try:
import base64
import io
import pybase64
from datasets import load_dataset
except ImportError:
raise ImportError("Please install datasets: pip install datasets")
@@ -867,7 +867,7 @@ def sample_mmmu_requests(
# Encode image to base64
buffered = io.BytesIO()
image.save(buffered, format="JPEG")
img_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
img_str = pybase64.b64encode(buffered.getvalue()).decode("utf-8")
image_data = f"data:image/jpeg;base64,{img_str}"
else:
continue