fix

arcee-ai · Jan 3, 2024 · b592919 · b592919
1 parent 4c99184
commit b592919
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mergekit/scripts/mixtral_moe.py b/mergekit/scripts/mixtral_moe.py
@@ -367,9 +367,9 @@ def main(
     build(
         config,
         out_path=out_path,
+        merge_options=merge_options,
         load_in_4bit=load_in_4bit,
         load_in_8bit=load_in_8bit,
-        lazy_unpickle=merge_options.lazy_unpickle,
         device=device,
     )