update download and model.generate
This commit is contained in:
parent
151b3aebf9
commit
74f07dd43c
@ -269,13 +269,14 @@ from modelscope import snapshot_download
|
|||||||
import os
|
import os
|
||||||
|
|
||||||
# download model in openxlab
|
# download model in openxlab
|
||||||
download(model_repo='ajupyter/EmoLLM_internlm2_7b_full',
|
# download(model_repo='MrCat/Meta-Llama-3-8B-Instruct',
|
||||||
output='ajupyter/EmoLLM_internlm2_7b_full')
|
# output='MrCat/Meta-Llama-3-8B-Instruct')
|
||||||
model_name_or_path = 'ajupyter/EmoLLM_internlm2_7b_full'
|
# model_name_or_path = 'MrCat/Meta-Llama-3-8B-Instruct'
|
||||||
|
|
||||||
# # download model in modelscope
|
# # download model in modelscope
|
||||||
# model_name_or_path = snapshot_download('ajupyter/EmoLLM_internlm2_7b_full',
|
# model_name_or_path = snapshot_download('LLM-Research/Meta-Llama-3-8B-Instruct',
|
||||||
# cache_dir='EmoLLM_internlm2_7b_full')
|
# cache_dir='LLM-Research/Meta-Llama-3-8B-Instruct')
|
||||||
|
|
||||||
|
|
||||||
# offline model
|
# offline model
|
||||||
model_name_or_path = "/root/EmoLLM/xtuner_config/merged_Llama"
|
model_name_or_path = "/root/EmoLLM/xtuner_config/merged_Llama"
|
||||||
@ -295,7 +296,7 @@ while True:
|
|||||||
input_text.replace(' ', '')
|
input_text.replace(' ', '')
|
||||||
if input_text == "exit":
|
if input_text == "exit":
|
||||||
break
|
break
|
||||||
response, history = model.chat(tokenizer, input_text, history=messages)
|
response, history = model.generate(tokenizer, input_text, history=messages)
|
||||||
messages.append((input_text, response))
|
messages.append((input_text, response))
|
||||||
print(f"robot >>> {response}")
|
print(f"robot >>> {response}")
|
||||||
```
|
```
|
||||||
|
Loading…
Reference in New Issue
Block a user