search

Eladlev / AutoPrompt

A framework for prompt tuning using Intent-based Prompt Calibration
Apache License 2.0
1.86k stars 149 forks source link

how to support qwen #63

Open xyhuo7788 opened 3 weeks ago

xyhuo7788 commented 3 weeks ago

use qwen-1.5 model command:

python run_pipeline.py --prompt "Does this movie review contain a spoiler? answer Yes or No" --task_description " Your task is to check if a movie review contains spoilers for the movie"

发生异常: KeyError 'samples' File "/mnt/AutoPrompt-main/optimization_pipeline.py", line 189, in samples_list = [element for sublist in samples_batches for element in sublist['samples']] File "/mnt/AutoPrompt-main/optimization_pipeline.py", line 189, in generate_initial_samples samples_list = [element for sublist in samples_batches for element in sublist['samples']] File "/mnt/AutoPrompt-main/optimization_pipeline.py", line 228, in step self.generate_initial_samples() File "/mnt/AutoPrompt-main/optimization_pipeline.py", line 255, in run_pipeline stop_criteria = self.step() File "/mnt/AutoPrompt-main/run_generation_pipeline.py", line 51, in best_prompt = ranker_pipeline.run_pipeline(opt.num_ranker_steps) KeyError: 'samples'

when i debug the code ,config file are all default,there is no key "sample", just only have "num_samples","task_description",'instruction',"text" is there anything wrong?

Eladlev commented 3 weeks ago

Hi, did you replace

meta_prompts:
    folder: 'prompts/meta_prompts_classification'

with: prompts/meta_prompts_completion ?