Skip to content

GitLab

Commit d8d15e4e authored by chenzk's avatar chenzk
Browse files

Update url.md

parent a613bbe5
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 10 deletions
README.md
View file @ d8d15e4e
...@@ -112,7 +112,7 @@ return True ...@@ -112,7 +112,7 @@ return True
## 数据集 ## 数据集
### 训练数据集 ### 训练数据集
VTimeLLM可基于Vicuna v1.5训练英文版本、基于ChatGLM3-6b训练中文版本，训练某个版本时只下载对应的数据集（data不同）即可。 VTimeLLM可基于Vicuna v1.5训练英文版本、基于ChatGLM3-6b训练中文版本，训练某个版本时只下载对应的数据集（data不同）即可。
训练数据集包括三阶段的数据集data和预提取特征feat两部分，完整数据集可通过[scnet](http://113.200.138.88:18080/aidatasets/project-dependency/vtimellm) 或官网链接进行下载。官网链接如下： 训练数据集包括三阶段的数据集data和预提取特征feat两部分，完整数据集可通过官网链接进行下载。官网链接如下：
ps：本仓库内准备了小数据集供训练测试，位于VTimeLLM/feat、VTimeLLM/data。 ps：本仓库内准备了小数据集供训练测试，位于VTimeLLM/feat、VTimeLLM/data。
...@@ -157,11 +157,9 @@ VTimeLLM可基于Vicuna v1.5训练英文版本、基于ChatGLM3-6b训练中文 ...@@ -157,11 +157,9 @@ VTimeLLM可基于Vicuna v1.5训练英文版本、基于ChatGLM3-6b训练中文
训练需要分别下载clip、Vicuna v1.5(或ChatGLM3-6b)权重，并将它们放入 'checkpoints' 目录中，下载链接如下： 训练需要分别下载clip、Vicuna v1.5(或ChatGLM3-6b)权重，并将它们放入 'checkpoints' 目录中，下载链接如下：
1、下载cilp模型 1、下载cilp模型
* [scnet](http://113.200.138.88:18080/aimodels/findsource-dependency/vtimellm)
* [官网链接](https://cloud.tsinghua.edu.cn/d/6db5d02883124826aa6f/?p=%2Fcheckpoints&mode=list) * [官网链接](https://cloud.tsinghua.edu.cn/d/6db5d02883124826aa6f/?p=%2Fcheckpoints&mode=list)
2-1、下载Vicuna v1.5权重 2-1、下载Vicuna v1.5权重
* [scnet](http://113.200.138.88:18080/aimodels/vicuna-7b-v1.5)
* [官网链接](https://huggingface.co/lmsys/vicuna-7b-v1.5/tree/main) * [官网链接](https://huggingface.co/lmsys/vicuna-7b-v1.5/tree/main)
* 代码下载（huggingface） * 代码下载（huggingface）
``` ```
...@@ -171,7 +169,6 @@ export HF_DATASETS_CACHE="./checkpoints/vicuna-7b-v1.5" ...@@ -171,7 +169,6 @@ export HF_DATASETS_CACHE="./checkpoints/vicuna-7b-v1.5"
huggingface-cli download --resume-download lmsys/vicuna-7b-v1.5 --local-dir checkpoints/vicuna-7b-v1.5 --local-dir-use-symlinks False huggingface-cli download --resume-download lmsys/vicuna-7b-v1.5 --local-dir checkpoints/vicuna-7b-v1.5 --local-dir-use-symlinks False
``` ```
2-2、下载ChatGLM3-6b权重 2-2、下载ChatGLM3-6b权重
* [scnet](http://113.200.138.88:18080/aimodels/chatglm3-6b)
* [官网链接](https://huggingface.co/THUDM/chatglm3-6b/tree/main) * [官网链接](https://huggingface.co/THUDM/chatglm3-6b/tree/main)
* 代码下载（huggingface） * 代码下载（huggingface）
``` ```
...@@ -202,7 +199,6 @@ sh scripts/stage3.sh ...@@ -202,7 +199,6 @@ sh scripts/stage3.sh
## 推理 ## 推理
VTimeLLM基于Vicuna v1.5训练了英文版本，存储为vtimellm-vicuna-v1-5-7b.tar.gz；基于ChatGLM3-6b训练了中文版本，存储为vtimellm-chatglm3-6b.tar.gz。推理某个版本时只下载对应的模型即可。 VTimeLLM基于Vicuna v1.5训练了英文版本，存储为vtimellm-vicuna-v1-5-7b.tar.gz；基于ChatGLM3-6b训练了中文版本，存储为vtimellm-chatglm3-6b.tar.gz。推理某个版本时只下载对应的模型即可。
推理需要分别下载clip、Vicuna v1.5（或ChatGLM3-6b）、VTimeLLM权重，并将它们放入 'checkpoints' 目录中。clip、Vicuna v1.5（或ChatGLM3-6b）的下载参考训练阶段，VTimeLLM权重的下载链接如下： 推理需要分别下载clip、Vicuna v1.5（或ChatGLM3-6b）、VTimeLLM权重，并将它们放入 'checkpoints' 目录中。clip、Vicuna v1.5（或ChatGLM3-6b）的下载参考训练阶段，VTimeLLM权重的下载链接如下：
* [scnet](http://113.200.138.88:18080/aimodels/findsource-dependency/vtimellm)
* [官网链接](https://cloud.tsinghua.edu.cn/d/6db5d02883124826aa6f/?p=%2Fcheckpoints&mode=list) * [官网链接](https://cloud.tsinghua.edu.cn/d/6db5d02883124826aa6f/?p=%2Fcheckpoints&mode=list)
解压VTimeLLM权重的代码如下： 解压VTimeLLM权重的代码如下：
...@@ -256,11 +252,6 @@ HIP_VISIBLE_DEVICES=0 python -m vtimellm.inference \ ...@@ -256,11 +252,6 @@ HIP_VISIBLE_DEVICES=0 python -m vtimellm.inference \
### 热点应用行业 ### 热点应用行业
`家具,电商,医疗,广媒,教育` `家具,电商,医疗,广媒,教育`
## 预训练权重 ## 预训练权重
- http://113.200.138.88:18080/aimodels/findsource-dependency/vtimellm (vtimellm、clip)
http://113.200.138.88:18080/aimodels/vicuna-7b-v1.5.git (vicuna-7b-v1.5)
http://113.200.138.88:18080/aimodels/chatglm3-6b (chatglm3-6b)
- https://cloud.tsinghua.edu.cn/d/6db5d02883124826aa6f/?p=%2Fcheckpoints&mode=list (vtimellm、clip) - https://cloud.tsinghua.edu.cn/d/6db5d02883124826aa6f/?p=%2Fcheckpoints&mode=list (vtimellm、clip)
https://huggingface.co/lmsys/vicuna-7b-v1.5/tree/main (vicuna-7b-v1.5) https://huggingface.co/lmsys/vicuna-7b-v1.5/tree/main (vicuna-7b-v1.5)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment