Allow both ROME and MEMIT to use cache

kmeng01 · kmeng01 · commit 5ad5ef171929 · 2022-11-08T18:12:21.000-05:00
diff --git a/experiments/evaluate.py b/experiments/evaluate.py
@@ -140,7 +140,7 @@ def main(
             if conserve_memory
             else dict()
         )
-        etc_args = dict(cache_template=cache_template) if "ROME" in alg_name else dict()
+        etc_args = dict(cache_template=cache_template) if any(alg in alg_name for alg in ["ROME", "MEMIT"]) else dict()
 
         start = time()
         edited_model, weights_copy = apply_algo(

Original file line number	Diff line number	Diff line change
`@@ -140,7 +140,7 @@ def main(`
`140`	`140`	`if conserve_memory`
`141`	`141`	`else dict()`
`142`	`142`	`)`
`143`		`- etc_args = dict(cache_template=cache_template) if "ROME" in alg_name else dict()`
	`143`	`+ etc_args = dict(cache_template=cache_template) if any(alg in alg_name for alg in ["ROME", "MEMIT"]) else dict()`
`144`	`144`
`145`	`145`	`start = time()`
`146`	`146`	`edited_model, weights_copy = apply_algo(`