Skip to content

Commit

Permalink
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
remove comments args
Browse files Browse the repository at this point in the history
sixsixcoder committed Oct 10, 2024
1 parent 2c89931 commit 13f675d
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions tests/models/decoder_only/vision_language/test_glm4.py
Original file line number Diff line number Diff line change
@@ -38,9 +38,7 @@ def run_test(
limit_mm_per_prompt={"image": mm_limit},
tensor_parallel_size=tensor_parallel_size,
distributed_executor_backend=distributed_executor_backend,
# gpu_memory_utilization=0.9,
enforce_eager=True) as vllm_model:
# tokenizer = vllm_model.model.get_tokenizer()
stop_token_ids = [151329, 151336, 151338]
vllm_outputs_per_image = [
vllm_model.generate_greedy_logprobs(prompts,
@@ -58,7 +56,6 @@ def run_test(
max_tokens,
num_logprobs=num_logprobs,
images=images,
# tokenizer=tokenizer
)
for prompts, images in inputs
]

0 comments on commit 13f675d

Please sign in to comment.