From 482c02982b0481eb63abe0b7e3315e6771ee5312 Mon Sep 17 00:00:00 2001 From: soumik12345 <19soumik.rakshit96@gmail.com> Date: Sat, 3 Aug 2024 11:01:05 +0000 Subject: [PATCH] update: examples --- .../evaluate_mllm_metric_complex.py | 13 ++----------- 1 file changed, 2 insertions(+), 11 deletions(-) diff --git a/examples/multimodal_llm_eval/evaluate_mllm_metric_complex.py b/examples/multimodal_llm_eval/evaluate_mllm_metric_complex.py index a9417ee..3e676f1 100644 --- a/examples/multimodal_llm_eval/evaluate_mllm_metric_complex.py +++ b/examples/multimodal_llm_eval/evaluate_mllm_metric_complex.py @@ -21,17 +21,8 @@ def main( wandb.init(project=project, entity=entity, job_type="evaluation") weave.init(project_name=project) - # dataset = weave.ref(dataset_ref).get() - # dataset = dataset.rows[:dataset_limit] if dataset_limit else dataset - dataset = ( - [ - {"prompt": "The black chair is on the right of the wooden table."}, - {"prompt": "The round clock was mounted on the white wall."}, - {"prompt": "The fluffy pillow was on the left of the striped blanket."}, - ] - if dataset_limit - else weave.ref(dataset_ref).get() - ) + dataset = weave.ref(dataset_ref).get() + dataset = dataset.rows[:dataset_limit] if dataset_limit else dataset diffusion_model = BaseDiffusionModel( diffusion_model_name_or_path=diffusion_model_address,