-
Notifications
You must be signed in to change notification settings - Fork 35
Open
Description
In Step 4 of installation instruction, what should I run python src/yue/infer.py --stage1_use_exl2 --stage2_use_exl2 --stage2_cache_size 32768 [original args]
what does [original args]
means.
if I run just with the lyrics and genre it shows AssertionError: Can't find m-a-p/YuE-s1-7B-anneal-en-cot
same issue with
python src/yue/infer.py --stage1_use_exl2 --stage2_use_exl2 --stage2_cache_size 32768 --cuda_idx 0 --stage1_model m-a-p/YuE-s1-7B-anneal-en-cot --stage2_model m-a-p/YuE-s2-1B-general --genre_txt ./prompt_egs/genre.txt --lyrics_txt ./prompt_egs/lyrics.txt --run_n_segments 2 --stage2_batch_size 4 --output_dir ./output --max_new_tokens 3000 --repetition_penalty 1.1
brcisna
Metadata
Metadata
Assignees
Labels
No labels