Skip to content

Update demo.py to fix the override of llama_type#141

Open
persistz wants to merge 1 commit intoOpenGVLab:mainfrom
persistz:patch-1
Open

Update demo.py to fix the override of llama_type#141
persistz wants to merge 1 commit intoOpenGVLab:mainfrom
persistz:patch-1

Conversation

@persistz
Copy link

As the definition of the function load, need to specify the parameter name of device.

def load(name, llama_dir, llama_type="7B", device="xxx", download_root='ckpts', max_seq_len=512, phase="finetune")

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant