Increase weight of generation time metric in FLUX

womboai · Nov 30, 2024 · 510d8e0 · 510d8e0
1 parent 849348b
commit 510d8e0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/base/base/contest.py b/base/base/contest.py
@@ -117,8 +117,8 @@ def normalize(baseline_value: float, benchmark_value: float, metric_type: Metric
         output_comparator=partial(ImageOutputComparator, CUDA_4090_DEVICE),
         baseline_repository=RepositoryInfo(url="https://github.com/womboai/flux-schnell-edge-inference", revision="fbfb8f0"),
         metric_weights={
-            MetricType.SIMILARITY_SCORE: 3,
-            MetricType.VRAM_USED: 3,
+            MetricType.SIMILARITY_SCORE: 1,
+            MetricType.VRAM_USED: 1,
             MetricType.GENERATION_TIME: 1,
         }
     ),