Skip to content

Instruction unclear #16

@uzairali001

Description

@uzairali001

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
Image

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions