ci: reduce and refactor vlm ut and combine test files (#11062)
This commit is contained in:
@@ -137,12 +137,10 @@ def generate_markdown_report(trace_dir, results: List["BenchmarkResult"]) -> str
|
||||
|
||||
# all results should share the same isl & osl
|
||||
for result in results:
|
||||
base_url = os.getenv(
|
||||
"TRACE_BASE_URL", "https://github.com/sgl-project/ci-data/traces"
|
||||
).rstrip("/")
|
||||
base_url = os.getenv("TRACE_BASE_URL", "").rstrip("/")
|
||||
relay_base = os.getenv(
|
||||
"PERFETTO_RELAY_URL",
|
||||
"https://docs.sglang.ai/ci-data/pages/perfetto_relay.html",
|
||||
"",
|
||||
).rstrip("/")
|
||||
summary += result.to_markdown_row(trace_dir, base_url, relay_base)
|
||||
|
||||
|
||||
@@ -901,7 +901,7 @@ class MllamaForConditionalGeneration(nn.Module):
|
||||
img = pixel_values[0, j]
|
||||
num_tiles = img.shape[0]
|
||||
batched_images[i, j, :num_tiles] = img
|
||||
batched_ar_ids[i, j] = mm_input.mm_items[0].aspect_ratio_id[0, j]
|
||||
batched_ar_ids[i, j] = mm_input.mm_items[0].aspect_ratio_ids[0, j]
|
||||
|
||||
batched_ar_mask[i, j, :num_tiles] = mm_input.mm_items[
|
||||
0
|
||||
|
||||
Reference in New Issue
Block a user