Re-Include compacter/compacter++

adapter-hub · Oct 29, 2023 · aec4eba · aec4eba
1 parent 4a88931
commit aec4eba
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/back_comp/Utils.py b/back_comp/Utils.py
@@ -99,8 +99,8 @@ def get_new_adapter_config_strings():
         "double_seq_bn_inv",
         "par_bn",
         "scaled_par_bn",
-        #"compacter",
-        #"compacter++",
+        "compacter",
+        "compacter++",
         "prefix_tuning",
         "prefix_tuning_flat",
         "lora",

diff --git a/back_comp/create_outputs.py b/back_comp/create_outputs.py
@@ -55,7 +55,7 @@
 
         # process & save output
         model_output_n, last_hidden_state = convert_tensors_to_list(model_output)
-        save_to_jsonl(model_output_n, adapter_config, os.path.join(model_dir, "output.jsonl"))
+        save_to_jsonl(model_output_n, config, os.path.join(model_dir, "output.jsonl"))
 
         # save adapter weights
         adapter_save_dir = os.path.join(model_dir, adapter_name)