Spaces:

ExplainabiliyForAATeam
/

explainability-tool-for-aa

Running

Milad Alshomary commited on Oct 23

Commit

779d850

1 Parent(s): 59c04b5

updates

Files changed (1) hide show

precompute_caches.py CHANGED Viewed

@@ -28,9 +28,9 @@ def precompute_all_caches(
     if models_to_test is None:
         models_to_test = [
-            # 'gabrielloiseau/LUAR-MUD-sentence-transformers',
-            # 'gabrielloiseau/LUAR-CRUD-sentence-transformers',
-            # 'miladalsh/light-luar',
             'AnnaWegmann/Style-Embedding'
         ]
@@ -197,7 +197,7 @@ if __name__ == "__main__":
     instances=[i for i in range(10)]  # First 20 instances for testing
     cache_stats = precompute_all_caches(
         models_to_test=[
-            'gabrielloiseau/LUAR-MUD-sentence-transformers'
         ],
         instances_to_process=instances
     )

     if models_to_test is None:
         models_to_test = [
+            'gabrielloiseau/LUAR-MUD-sentence-transformers',
+            'gabrielloiseau/LUAR-CRUD-sentence-transformers',
+            'miladalsh/light-luar',
             'AnnaWegmann/Style-Embedding'
         ]
     instances=[i for i in range(10)]  # First 20 instances for testing
     cache_stats = precompute_all_caches(
         models_to_test=[
+             'AnnaWegmann/Style-Embedding'
         ],
         instances_to_process=instances
     )