llama3.1
This commit is contained in:
006861
parent
7ffc91ff7d
commit
a8511861c1
177
app_Llama3_Gradio2.py
Normal file
177
app_Llama3_Gradio2.py
Normal file
@ -0,0 +1,177 @@
|
|||||||
|
import gradio as gr
|
||||||
|
import os
|
||||||
|
import torch
|
||||||
|
from transformers import GemmaTokenizer, AutoModelForCausalLM
|
||||||
|
from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
|
||||||
|
from threading import Thread
|
||||||
|
|
||||||
|
|
||||||
|
DESCRIPTION = '''
|
||||||
|
<div>
|
||||||
|
<h1 style="text-align: center;">EmoLLM Llama3 心理咨询室 V4.0</h1>
|
||||||
|
|
||||||
|
<p align="center">
|
||||||
|
<a href="https://github.com/SmartFlowAI/EmoLLM/">
|
||||||
|
<img src="https://st-app-center-006861-9746-jlroxvg.openxlab.space/media/cda6c1a05dc8ba5b19ad3e7a24920fdf3750c917751202385a6dbc51.png" alt="Logo" width="20%">
|
||||||
|
</a>
|
||||||
|
</p>
|
||||||
|
|
||||||
|
<div align="center">
|
||||||
|
|
||||||
|
<!-- PROJECT SHIELDS -->
|
||||||
|
[![OpenXLab_Model][OpenXLab_Model-image]][OpenXLab_Model-url]
|
||||||
|
|
||||||
|
<h2 style="text-align: center;"> EmoLLM是一系列能够支持 理解用户-支持用户-帮助用户 心理健康辅导链路的 心理健康大模型 ，欢迎大家star~⭐⭐</h2>
|
||||||
|
<p>https://github.com/SmartFlowAI/EmoLLM</p>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
</div>
|
||||||
|
|
||||||
|
[OpenXLab_Model-image]: https://cdn-static.openxlab.org.cn/header/openxlab_models.svg
|
||||||
|
[OpenXLab_Model-url]: https://openxlab.org.cn/models/detail/chg0901/EmoLLM-Llama3-8B-Instruct3.0
|
||||||
|
|
||||||
|
'''
|
||||||
|
|
||||||
|
LICENSE = """
|
||||||
|
<p align="center"> Built with Meta Llama 3 </>
|
||||||
|
"""
|
||||||
|
|
||||||
|
PLACEHOLDER = """
|
||||||
|
<div style="padding: 30px; text-align: center; display: flex; flex-direction: column; align-items: center;">
|
||||||
|
|
||||||
|
</div>
|
||||||
|
"""
|
||||||
|
|
||||||
|
|
||||||
|
css = """
|
||||||
|
h1 {
|
||||||
|
text-align: center;
|
||||||
|
display: block;
|
||||||
|
}
|
||||||
|
<!--
|
||||||
|
#duplicate-button {
|
||||||
|
margin: auto;
|
||||||
|
color: white;
|
||||||
|
background: #1565c0;
|
||||||
|
border-radius: 100vh;
|
||||||
|
}
|
||||||
|
-->
|
||||||
|
"""
|
||||||
|
|
||||||
|
# download internlm2 to the base_path directory using git tool
|
||||||
|
base_path = './EmoLLM-Llama3-8B-Instruct3.0'
|
||||||
|
os.system(f'git clone https://code.openxlab.org.cn/chg0901/EmoLLM-Llama3-8B-Instruct3.0.git {base_path}')
|
||||||
|
os.system(f'cd {base_path} && git lfs pull')
|
||||||
|
|
||||||
|
|
||||||
|
# Load the tokenizer and model
|
||||||
|
tokenizer = AutoTokenizer.from_pretrained(base_path,trust_remote_code=True)
|
||||||
|
model = AutoModelForCausalLM.from_pretrained(base_path,trust_remote_code=True, device_map="auto", torch_dtype=torch.float16).eval() # to("cuda:0")
|
||||||
|
terminators = [
|
||||||
|
tokenizer.eos_token_id,
|
||||||
|
tokenizer.convert_tokens_to_ids("<|eot_id|>")
|
||||||
|
]
|
||||||
|
|
||||||
|
def chat_llama3_8b(message: str,
|
||||||
|
history: list,
|
||||||
|
temperature: float,
|
||||||
|
max_new_tokens: int,
|
||||||
|
top_p: float
|
||||||
|
) -> str:
|
||||||
|
"""
|
||||||
|
Generate a streaming response using the llama3-8b model.
|
||||||
|
Args:
|
||||||
|
message (str): The input message.
|
||||||
|
history (list): The conversation history used by ChatInterface.
|
||||||
|
temperature (float): The temperature for generating the response.
|
||||||
|
max_new_tokens (int): The maximum number of new tokens to generate.
|
||||||
|
Returns:
|
||||||
|
str: The generated response.
|
||||||
|
"""
|
||||||
|
conversation = []
|
||||||
|
|
||||||
|
for user, assistant in history:
|
||||||
|
conversation.extend([{"role": "user", "content": user}, {"role": "assistant", "content": assistant}])
|
||||||
|
conversation.append({"role": "user", "content": message})
|
||||||
|
|
||||||
|
input_ids = tokenizer.apply_chat_template(conversation, return_tensors="pt").to(model.device)
|
||||||
|
|
||||||
|
streamer = TextIteratorStreamer(tokenizer, timeout=10.0, skip_prompt=True, skip_special_tokens=True)
|
||||||
|
|
||||||
|
generate_kwargs = dict(
|
||||||
|
input_ids= input_ids,
|
||||||
|
streamer=streamer,
|
||||||
|
max_new_tokens=max_new_tokens,
|
||||||
|
do_sample=True,
|
||||||
|
temperature=temperature,
|
||||||
|
top_p = top_p,
|
||||||
|
eos_token_id=terminators,
|
||||||
|
)
|
||||||
|
# This will enforce greedy generation (do_sample=False) when the temperature is passed 0, avoiding the crash.
|
||||||
|
if temperature == 0:
|
||||||
|
generate_kwargs['do_sample'] = False
|
||||||
|
|
||||||
|
t = Thread(target=model.generate, kwargs=generate_kwargs)
|
||||||
|
t.start()
|
||||||
|
|
||||||
|
outputs = []
|
||||||
|
for text in streamer:
|
||||||
|
outputs.append(text)
|
||||||
|
yield "".join(outputs)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
# Gradio block
|
||||||
|
chatbot=gr.Chatbot(height=450, placeholder=PLACEHOLDER, label='EmoLLM Chat')
|
||||||
|
|
||||||
|
with gr.Blocks(fill_height=True, css=css) as demo:
|
||||||
|
|
||||||
|
gr.Markdown(DESCRIPTION)
|
||||||
|
# gr.DuplicateButton(value="Duplicate Space for private use", elem_id="duplicate-button")
|
||||||
|
gr.ChatInterface(
|
||||||
|
fn=chat_llama3_8b,
|
||||||
|
chatbot=chatbot,
|
||||||
|
fill_height=True,
|
||||||
|
additional_inputs_accordion=gr.Accordion(label="⚙️ Parameters", open=False, render=False),
|
||||||
|
additional_inputs=[
|
||||||
|
gr.Slider(minimum=0,
|
||||||
|
maximum=1,
|
||||||
|
step=0.1,
|
||||||
|
value=0.95,
|
||||||
|
label="Temperature",
|
||||||
|
render=False),
|
||||||
|
gr.Slider(minimum=128,
|
||||||
|
maximum=4096,
|
||||||
|
step=1,
|
||||||
|
value=4096,
|
||||||
|
label="Max new tokens",
|
||||||
|
render=False ),
|
||||||
|
gr.Slider(minimum=0.0,
|
||||||
|
maximum=1,
|
||||||
|
step=0.01,
|
||||||
|
value=0.8,
|
||||||
|
label="Top P",
|
||||||
|
render=False ),
|
||||||
|
# gr.Slider(minimum=128,
|
||||||
|
# maximum=4096,
|
||||||
|
# step=1,
|
||||||
|
# value=512,
|
||||||
|
# label="Max new tokens",
|
||||||
|
# render=False ),
|
||||||
|
],
|
||||||
|
examples=[
|
||||||
|
['请介绍你自己。'],
|
||||||
|
['我觉得我在学校的学习压力好大啊，虽然我真的很喜欢我的专业，但最近总是担心自己无法达到自己的期望，这让我有点焦虑。'],
|
||||||
|
['我最近总觉得自己在感情上陷入了困境，我喜欢上了我的朋友，但又害怕表达出来会破坏我们现在的关系...'],
|
||||||
|
['我感觉自己像是被困在一个无尽的循环中。每天醒来都感到身体沉重，对日常活动提不起兴趣，工作、锻炼甚至是我曾经喜欢的事物都让我觉得厌倦'],
|
||||||
|
['最近工作压力特别大，还有一些家庭矛盾']
|
||||||
|
],
|
||||||
|
cache_examples=False,
|
||||||
|
)
|
||||||
|
|
||||||
|
gr.Markdown(LICENSE)
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
demo.launch()
|
||||||
|
|
||||||
|
|
||||||
@ -0,0 +1,243 @@
|
|||||||
|
# Copyright (c) OpenMMLab. All rights reserved.
|
||||||
|
"""
|
||||||
|
Ref: https://github.com/InternLM/xtuner/edit/main/xtuner/configs/internlm/internlm2_5_chat_7b/internlm2_5_chat_7b_full_finetune_custom_dataset_e1.py
|
||||||
|
|
||||||
|
Data format:
|
||||||
|
[
|
||||||
|
{
|
||||||
|
"conversation": [
|
||||||
|
{
|
||||||
|
"system": "",
|
||||||
|
"input": "xxx",
|
||||||
|
"output": "xxx"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"input": "xxx",
|
||||||
|
"output": "xxx"
|
||||||
|
}
|
||||||
|
]
|
||||||
|
},
|
||||||
|
...
|
||||||
|
]
|
||||||
|
Please refer to https://github.com/InternLM/xtuner/blob/main/docs/en/user_guides/dataset_format.md for details.
|
||||||
|
""" # noqa: E501
|
||||||
|
from datasets import load_dataset
|
||||||
|
from mmengine.hooks import (CheckpointHook, DistSamplerSeedHook, IterTimerHook,
|
||||||
|
LoggerHook, ParamSchedulerHook)
|
||||||
|
from mmengine.optim import AmpOptimWrapper, CosineAnnealingLR
|
||||||
|
from torch.optim import AdamW
|
||||||
|
from torch.utils.data import BatchSampler
|
||||||
|
from transformers import AutoModelForCausalLM, AutoTokenizer
|
||||||
|
|
||||||
|
from xtuner.dataset import process_hf_dataset
|
||||||
|
from xtuner.dataset.collate_fns import default_collate_fn
|
||||||
|
from xtuner.dataset.map_fns import template_map_fn_factory
|
||||||
|
from xtuner.dataset.samplers import InternRepoSampler
|
||||||
|
from xtuner.engine import (DatasetInfoHook, EvaluateChatHook, ThroughputHook,
|
||||||
|
VarlenAttnArgsToMessageHubHook)
|
||||||
|
from xtuner.engine.runner import TrainLoop
|
||||||
|
from xtuner.model import SupervisedFinetune
|
||||||
|
from xtuner.utils import PROMPT_TEMPLATE
|
||||||
|
|
||||||
|
#######################################################################
|
||||||
|
# PART 1 Settings #
|
||||||
|
#######################################################################
|
||||||
|
# Model
|
||||||
|
pretrained_model_name_or_path = '/root/group_share/Hong/Meta-Llama-3___1-8B-Instruct'
|
||||||
|
use_varlen_attn = True
|
||||||
|
|
||||||
|
# Data
|
||||||
|
data_files = ['/root/EmoLLM/datasets/multi_turn_dataset_2.json']
|
||||||
|
prompt_template = PROMPT_TEMPLATE.internlm2_chat
|
||||||
|
# max_length = 32768
|
||||||
|
max_length = int(32768/4) ## A100*2
|
||||||
|
pack_to_max_length = True
|
||||||
|
|
||||||
|
# parallel
|
||||||
|
sequence_parallel_size = 1
|
||||||
|
|
||||||
|
# Scheduler & Optimizer
|
||||||
|
# batch size per device, set to 1 if `use_varlen_attn` = True
|
||||||
|
# To clarify, enlarging the batch size essentially enlarges the `max_length`.
|
||||||
|
# For example, doubling the max length is tantamount to doubling the batch size
|
||||||
|
batch_size = 1
|
||||||
|
accumulative_counts = 1 # 1bs * 1acc * 64gpu = 64 batchsize
|
||||||
|
accumulative_counts *= sequence_parallel_size
|
||||||
|
dataloader_num_workers = 4
|
||||||
|
max_epochs = 3
|
||||||
|
|
||||||
|
optim_type = AdamW
|
||||||
|
lr = 4e-5
|
||||||
|
betas = (0.9, 0.95)
|
||||||
|
weight_decay = 0.01
|
||||||
|
max_norm = 1 # grad clip
|
||||||
|
warm_up_ratio = 0.025
|
||||||
|
|
||||||
|
# Save
|
||||||
|
save_steps = 500
|
||||||
|
save_total_limit = 2 # Maximum checkpoints to keep (-1 means unlimited)
|
||||||
|
|
||||||
|
# Evaluate the generation performance during the training
|
||||||
|
evaluation_freq = 500
|
||||||
|
SYSTEM = "你由EmoLLM团队打造的中文领域心理健康助手, 是一个研究过无数具有心理健康问题的病人与心理健康医生对话的心理专家, 在心理方面拥有广博的知识储备和丰富的研究咨询经验，你旨在通过专业心理咨询, 协助来访者完成心理诊断。请充分利用专业心理学知识与咨询技术, 一步步帮助来访者解决心理问题, 接下来你将只使用中文来回答和咨询问题。"
|
||||||
|
evaluation_inputs = [
|
||||||
|
# '躲在云朵下面就不怕淋雨了', # ruozhi train
|
||||||
|
# '李白如果告语文书侵权的话能赔不少钱吧', # ruozhi test
|
||||||
|
# '雨天，我走进水坑里，不小心踩碎了天空。', # ruozhi test
|
||||||
|
'请介绍你自己', # self cognition
|
||||||
|
'你好',
|
||||||
|
'我今天心情不好，感觉不开心，很烦。',
|
||||||
|
'我最近总是感到很焦虑，尤其是在学业上。我有个特别崇拜的同学，他好像在各方面都比我优秀，我总觉得自己怎么努力也追不上他，这让我压力特别大。',
|
||||||
|
]
|
||||||
|
|
||||||
|
# "这是一句富有想象力和幽默感的表达。在现实生活中，躲在云朵下面并不能避免淋雨，因为云朵实际上是由水蒸气凝结形成的，而雨就是由这些水滴凝结而成的。\n\n这样的表达可能是在夸张和幽默的语境中，通过一种天马行空的方式来表达逃避现实或者寻找避难的愿望。在文学或口语表达中，常常会运用夸张和幽默来传达情感或者引起共鸣。",
|
||||||
|
|
||||||
|
# "如果李白（701年2月28日—762年12月）真的能就侵权诉讼获得赔偿，那确实可能是一笔不小的金额。然而，这种想法主要是一种有趣的假设，因为现实中有几个关键因素使这种情况不可能发生：\n\n1. **时间差异**：李白生活在唐朝，距今大约1200多年前。那 个时代的法律体系与现代的知识产权法律截然不同，当时没有现代意义上的版权法。\n\n2. **版权法的适用范围**：即使在现代，版权 法也有一定的时效限制。在大多数国家，版权保护通常在作者去世后一定年数内有效（如我国是作者去世后50年）。李白去世已超过1250年，因此其作品已经进入公共领域，任何人都可以自由使用而无需支付版权费用。\n\n3. **历史与现实的区别**：历史人物无法在现代 法律体系中提起诉讼，因为他们不再是活跃的法律主体。\n\n所以，虽然这是一个有趣的想法，但在现实中，李白或其他古代作者无法因其作品被现代出版物使用而获得赔偿。",
|
||||||
|
|
||||||
|
# "这个描述似乎是一个修辞手法，比喻性地描述了雨天的场景。在这个描述中，说“我走进水坑里，不小心踩碎了天空”，实际上并非字面意义上的发生，而是一种用词语来比喻雨天的场景。\n\n通常情况下，当雨水落在水坑或者蓄水池时，水面会泛起涟漪或者波纹，可能会反射天空的颜色或者天空的倒影。因此，这句话可能是通过“踩碎了天空”的说法来比喻雨天时踩进水坑的情景，描述雨水落在水坑中形成的波纹或者涟漪，产生了一种倒映天空的效果。\n\n这种形象化的表达方式可能是为了更生动地描述一个平常的场景，赋予它一些诗意或者意境。",
|
||||||
|
|
||||||
|
#######################################################################
|
||||||
|
# PART 2 Model & Tokenizer #
|
||||||
|
#######################################################################
|
||||||
|
tokenizer = dict(
|
||||||
|
type=AutoTokenizer.from_pretrained,
|
||||||
|
pretrained_model_name_or_path=pretrained_model_name_or_path,
|
||||||
|
trust_remote_code=True,
|
||||||
|
padding_side='right')
|
||||||
|
|
||||||
|
model = dict(
|
||||||
|
type=SupervisedFinetune,
|
||||||
|
use_varlen_attn=use_varlen_attn,
|
||||||
|
llm=dict(
|
||||||
|
type=AutoModelForCausalLM.from_pretrained,
|
||||||
|
pretrained_model_name_or_path=pretrained_model_name_or_path,
|
||||||
|
trust_remote_code=True))
|
||||||
|
|
||||||
|
#######################################################################
|
||||||
|
# PART 3 Dataset & Dataloader #
|
||||||
|
#######################################################################
|
||||||
|
train_dataset = dict(
|
||||||
|
type=process_hf_dataset,
|
||||||
|
use_varlen_attn=use_varlen_attn,
|
||||||
|
dataset=dict(type=load_dataset, path='json', data_files=data_files),
|
||||||
|
tokenizer=tokenizer,
|
||||||
|
max_length=max_length,
|
||||||
|
dataset_map_fn=None,
|
||||||
|
template_map_fn=dict(
|
||||||
|
type=template_map_fn_factory, template=prompt_template),
|
||||||
|
remove_unused_columns=True,
|
||||||
|
shuffle_before_pack=True,
|
||||||
|
pack_to_max_length=pack_to_max_length)
|
||||||
|
|
||||||
|
train_dataloader = dict(
|
||||||
|
batch_size=batch_size,
|
||||||
|
num_workers=dataloader_num_workers,
|
||||||
|
dataset=train_dataset,
|
||||||
|
sampler=dict(type=InternRepoSampler, shuffle=True, seed=1024),
|
||||||
|
batch_sampler=dict(
|
||||||
|
type=BatchSampler, drop_last=True, batch_size=batch_size),
|
||||||
|
collate_fn=dict(type=default_collate_fn, use_varlen_attn=use_varlen_attn))
|
||||||
|
|
||||||
|
#######################################################################
|
||||||
|
# PART 4 Scheduler & Optimizer #
|
||||||
|
#######################################################################
|
||||||
|
# optimizer
|
||||||
|
optim_wrapper = dict(
|
||||||
|
type=AmpOptimWrapper,
|
||||||
|
optimizer=dict(
|
||||||
|
type=optim_type, lr=lr, betas=betas, weight_decay=weight_decay),
|
||||||
|
clip_grad=dict(max_norm=max_norm, error_if_nonfinite=False),
|
||||||
|
accumulative_counts=accumulative_counts,
|
||||||
|
loss_scale='dynamic',
|
||||||
|
)
|
||||||
|
|
||||||
|
# learning policy
|
||||||
|
# More information: https://github.com/open-mmlab/mmengine/blob/main/docs/en/tutorials/param_scheduler.md # noqa: E501
|
||||||
|
param_scheduler = [
|
||||||
|
dict(
|
||||||
|
type='LinearLR',
|
||||||
|
start_factor=1 / 40,
|
||||||
|
by_epoch=True,
|
||||||
|
begin=0,
|
||||||
|
end=warm_up_ratio * max_epochs,
|
||||||
|
convert_to_iter_based=True),
|
||||||
|
dict(
|
||||||
|
type=CosineAnnealingLR,
|
||||||
|
eta_min=lr * 0.15,
|
||||||
|
by_epoch=True,
|
||||||
|
begin=warm_up_ratio * max_epochs,
|
||||||
|
end=max_epochs,
|
||||||
|
convert_to_iter_based=True)
|
||||||
|
]
|
||||||
|
|
||||||
|
# train, val, test setting
|
||||||
|
train_cfg = dict(type=TrainLoop, max_epochs=max_epochs)
|
||||||
|
|
||||||
|
#######################################################################
|
||||||
|
# PART 5 Runtime #
|
||||||
|
#######################################################################
|
||||||
|
# Log the dialogue periodically during the training process, optional
|
||||||
|
custom_hooks = [
|
||||||
|
dict(
|
||||||
|
type=DatasetInfoHook, tokenizer=tokenizer,
|
||||||
|
is_intern_repo_dataset=True),
|
||||||
|
dict(
|
||||||
|
type=EvaluateChatHook,
|
||||||
|
tokenizer=tokenizer,
|
||||||
|
every_n_iters=evaluation_freq,
|
||||||
|
evaluation_inputs=evaluation_inputs,
|
||||||
|
system=SYSTEM,
|
||||||
|
prompt_template=prompt_template),
|
||||||
|
dict(type=ThroughputHook)
|
||||||
|
]
|
||||||
|
|
||||||
|
if use_varlen_attn:
|
||||||
|
custom_hooks += [dict(type=VarlenAttnArgsToMessageHubHook)]
|
||||||
|
|
||||||
|
# configure default hooks
|
||||||
|
default_hooks = dict(
|
||||||
|
# record the time of every iteration.
|
||||||
|
timer=dict(type=IterTimerHook),
|
||||||
|
# print log every 100 iterations.
|
||||||
|
logger=dict(type=LoggerHook, log_metric_by_epoch=False, interval=1),
|
||||||
|
# enable the parameter scheduler.
|
||||||
|
param_scheduler=dict(type=ParamSchedulerHook),
|
||||||
|
# save checkpoint per `save_steps`.
|
||||||
|
checkpoint=dict(
|
||||||
|
type=CheckpointHook,
|
||||||
|
by_epoch=False,
|
||||||
|
interval=save_steps,
|
||||||
|
max_keep_ckpts=save_total_limit),
|
||||||
|
# set sampler seed in distributed evrionment.
|
||||||
|
sampler_seed=dict(type=DistSamplerSeedHook),
|
||||||
|
)
|
||||||
|
|
||||||
|
# configure environment
|
||||||
|
env_cfg = dict(
|
||||||
|
# whether to enable cudnn benchmark
|
||||||
|
cudnn_benchmark=False,
|
||||||
|
# set multi process parameters
|
||||||
|
mp_cfg=dict(mp_start_method='fork', opencv_num_threads=0),
|
||||||
|
# set distributed parameters
|
||||||
|
dist_cfg=dict(backend='nccl'),
|
||||||
|
)
|
||||||
|
|
||||||
|
# set visualizer
|
||||||
|
visualizer = None
|
||||||
|
|
||||||
|
# set log level
|
||||||
|
log_level = 'INFO'
|
||||||
|
|
||||||
|
# load from which checkpoint
|
||||||
|
load_from = None
|
||||||
|
|
||||||
|
# whether to resume training from the loaded checkpoint
|
||||||
|
resume = False
|
||||||
|
|
||||||
|
# Defaults to use random seed and disable `deterministic`
|
||||||
|
randomness = dict(seed=None, deterministic=False)
|
||||||
|
|
||||||
|
log_processor = dict(
|
||||||
|
by_epoch=False,
|
||||||
|
window_size=1,
|
||||||
|
mean_pattern=r'.*(loss|time|data_time|grad_norm|tflops).*')
|
||||||
Loading…
Reference in New Issue
Block a user