From 13e0eab671e5a1b0935a6dfb1babf905c16678d7 Mon Sep 17 00:00:00 2001 From: Xianchao Wu Date: Wed, 16 Aug 2023 06:10:23 +0000 Subject: [PATCH] dgx1 change --- WizardCoder/mbpp_1.sh | 6 +++--- WizardCoder/src/mbpp_gen.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/WizardCoder/mbpp_1.sh b/WizardCoder/mbpp_1.sh index 9a8f4d3..7b93a31 100644 --- a/WizardCoder/mbpp_1.sh +++ b/WizardCoder/mbpp_1.sh @@ -9,9 +9,9 @@ #model="/path/to/your/model" model="/workspace/asr/WizardLM/WizardCoder/models--WizardLM--WizardCoder-15B-V1.0/snapshots/926ca1b215c4631bc5f8c3e47173381452c23e5c" temp=0.2 # temperature, TODO reset this 温度 -max_len=2048 -pred_num=200 -num_seqs_per_iter=2 +max_len=1024 #2048 +pred_num=1 #200 TODO +num_seqs_per_iter=1 #2 output_path=preds/MBPP_T${temp}_N${pred_num} mbpp_path=data/mbpp.test.jsonl # we provide this file in data/mbpp.test.zip diff --git a/WizardCoder/src/mbpp_gen.py b/WizardCoder/src/mbpp_gen.py index d358603..3647b09 100644 --- a/WizardCoder/src/mbpp_gen.py +++ b/WizardCoder/src/mbpp_gen.py @@ -53,7 +53,7 @@ def generate_prompt(input): return INSTRUCTION # e.g., 'Below is an instruction that describes a task. Write a response that appropriately completes the request.\n\n### Instruction:\nCreate a Python script for this problem:\n\nWrite a python function to remove first and last occurrence of a given character from the string.\nTest examples:\nassert remove_Occ("hello","l") == "heo"\nassert remove_Occ("abcda","a") == "bcd"\nassert remove_Occ("PHP","P") == "H"\n\n### Response:' def get_model( - load_8bit: bool = False, + load_8bit: bool = False, # TODO for GPU memory saving base_model: str = "bigcode/starcoder", ): assert base_model, (