Update utils.py

FreedomIntelligence · Sep 14, 2024 · b6a841c · b6a841c
1 parent 35c34a3
commit b6a841c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/proxy-tuning/eval/utils.py b/src/proxy-tuning/eval/utils.py
@@ -84,7 +84,7 @@ def generate_completions(
 
     num_return_sequences = generation_kwargs.get("num_return_sequences", 1)
     model_type = model.base.config.model_type
-    output_path = generation_kwargs.get("output_path", "/223040239/medbase/src/proxy-tuning/outputs/gemma-7b.jsonl")
+    output_path = generation_kwargs.get("output_path", "../outputs/gemma-7b.jsonl")
     fp = open(output_path,'w')
     for i in range(0, len(prompts), batch_size):
         batch_items = prompts[i:i+batch_size]
@@ -380,4 +380,4 @@ def dynamic_import_function(function_path):
     module_path, function_name = function_path.rsplit(".", 1)
     module = import_module(module_path)
     function = getattr(module, function_name)
-    return function
+    return function