diff --git a/README.md b/README.md index 959d46ec311c88ec0786d6525e9e8d1f7a9619a6..494ad536c2bd3566e88f8a2ef45c5f4b70455075 100644 --- a/README.md +++ b/README.md @@ -43,9 +43,9 @@ DeepSparkHub甄选上百个应用算法和模型，覆盖AI和通用计算各领 | [QWen1.5-7B](nlp/llm/qwen1.5-7b/pytorch) | PyTorch | Firefly | school_math | 4.1.1 | | [QWen1.5-14B](nlp/llm/qwen1.5-14b/pytorch) | PyTorch | Firefly | school_math | 4.1.1 | | [Qwen2.5-7B SFT](nlp/llm/qwen2.5-7b/pytorch) | PyTorch | LLaMA-Factory | qwen2.5-7b | 4.1.1 | -| [Yi-6B](nlp/llm/Yi-6B/pytorch) | PyTorch | DeepSpeed | Yi-6B | 4.2.0 | -| [Yi-1.5-6B](nlp/llm/Yi-1.5-6B/pytorch) | PyTorch | DeepSpeed | Yi-1.5-6B | 4.2.0 | -| [Yi-VL-6B](nlp/llm/Yi-VL-6B/pytorch) | PyTorch | LLaMA-Factory | Yi-VL-6B-hf | 4.2.0 | +| [Yi-6B](nlp/llm/yi-6b/pytorch) | PyTorch | DeepSpeed | Yi-6B | 4.2.0 | +| [Yi-1.5-6B](nlp/llm/yi-1.5-6b/pytorch) | PyTorch | DeepSpeed | Yi-1.5-6B | 4.2.0 | +| [Yi-VL-6B](nlp/llm/yi-vl-6b/pytorch) | PyTorch | LLaMA-Factory | Yi-VL-6B-hf | 4.2.0 | ### 计算机视觉（CV） diff --git a/README_en.md b/README_en.md index 21defa80f98072b79cff0b35119e1597b2b8d5a2..b8491f12d65200f62298c858c35e34f43c0e8097 100644 --- a/README_en.md +++ b/README_en.md @@ -45,9 +45,9 @@ individuals, healthcare, education, communication, energy, and more. | [QWen1.5-7B](nlp/llm/qwen1.5-7b/pytorch) | PyTorch | Firefly | school_math | 4.1.1 | | [QWen1.5-14B](nlp/llm/qwen1.5-14b/pytorch) | PyTorch | Firefly | school_math | 4.1.1 | | [Qwen2.5-7B SFT](nlp/llm/qwen2.5-7b/pytorch) | PyTorch | LLaMA-Factory | qwen2.5-7b | 4.1.1 | -| [Yi-6B](nlp/llm/Yi-6B/pytorch) | PyTorch | DeepSpeed | Yi-6B | 4.2.0 | -| [Yi-1.5-6B](nlp/llm/Yi-1.5-6B/pytorch) | PyTorch | DeepSpeed | Yi-1.5-6B | 4.2.0 | -| [Yi-VL-6B](nlp/llm/Yi-VL-6B/pytorch) | PyTorch | LLaMA-Factory | Yi-VL-6B-hf | 4.2.0 | +| [Yi-6B](nlp/llm/yi-6b/pytorch) | PyTorch | DeepSpeed | Yi-6B | 4.2.0 | +| [Yi-1.5-6B](nlp/llm/yi-1.5-6b/pytorch) | PyTorch | DeepSpeed | Yi-1.5-6B | 4.2.0 | +| [Yi-VL-6B](nlp/llm/yi-vl-6b/pytorch) | PyTorch | LLaMA-Factory | Yi-VL-6B-hf | 4.2.0 | ### Computer Vision diff --git a/nlp/llm/Yi-1.5-6B/pytorch/README.md b/nlp/llm/yi-1.5-6b/pytorch/README.md similarity index 100% rename from nlp/llm/Yi-1.5-6B/pytorch/README.md rename to nlp/llm/yi-1.5-6b/pytorch/README.md diff --git a/nlp/llm/Yi-1.5-6B/pytorch/constant.py b/nlp/llm/yi-1.5-6b/pytorch/constant.py similarity index 100% rename from nlp/llm/Yi-1.5-6B/pytorch/constant.py rename to nlp/llm/yi-1.5-6b/pytorch/constant.py diff --git a/nlp/llm/Yi-1.5-6B/pytorch/requirements.txt b/nlp/llm/yi-1.5-6b/pytorch/requirements.txt similarity index 100% rename from nlp/llm/Yi-1.5-6B/pytorch/requirements.txt rename to nlp/llm/yi-1.5-6b/pytorch/requirements.txt diff --git a/nlp/llm/Yi-1.5-6B/pytorch/scripts/run_sft_Yi-1.5-6b.sh b/nlp/llm/yi-1.5-6b/pytorch/scripts/run_sft_Yi-1.5-6b.sh similarity index 100% rename from nlp/llm/Yi-1.5-6B/pytorch/scripts/run_sft_Yi-1.5-6b.sh rename to nlp/llm/yi-1.5-6b/pytorch/scripts/run_sft_Yi-1.5-6b.sh diff --git a/nlp/llm/Yi-1.5-6B/pytorch/sft/main.py b/nlp/llm/yi-1.5-6b/pytorch/sft/main.py similarity index 100% rename from nlp/llm/Yi-1.5-6B/pytorch/sft/main.py rename to nlp/llm/yi-1.5-6b/pytorch/sft/main.py diff --git a/nlp/llm/Yi-1.5-6B/pytorch/sft/prompt_eval.py b/nlp/llm/yi-1.5-6b/pytorch/sft/prompt_eval.py similarity index 100% rename from nlp/llm/Yi-1.5-6B/pytorch/sft/prompt_eval.py rename to nlp/llm/yi-1.5-6b/pytorch/sft/prompt_eval.py diff --git a/nlp/llm/Yi-1.5-6B/pytorch/utils/data/data_utils.py b/nlp/llm/yi-1.5-6b/pytorch/utils/data/data_utils.py similarity index 100% rename from nlp/llm/Yi-1.5-6B/pytorch/utils/data/data_utils.py rename to nlp/llm/yi-1.5-6b/pytorch/utils/data/data_utils.py diff --git a/nlp/llm/Yi-1.5-6B/pytorch/utils/data/raw_datasets.py b/nlp/llm/yi-1.5-6b/pytorch/utils/data/raw_datasets.py similarity index 100% rename from nlp/llm/Yi-1.5-6B/pytorch/utils/data/raw_datasets.py rename to nlp/llm/yi-1.5-6b/pytorch/utils/data/raw_datasets.py diff --git a/nlp/llm/Yi-1.5-6B/pytorch/utils/ds_utils.py b/nlp/llm/yi-1.5-6b/pytorch/utils/ds_utils.py similarity index 100% rename from nlp/llm/Yi-1.5-6B/pytorch/utils/ds_utils.py rename to nlp/llm/yi-1.5-6b/pytorch/utils/ds_utils.py diff --git a/nlp/llm/Yi-1.5-6B/pytorch/utils/model/model_utils.py b/nlp/llm/yi-1.5-6b/pytorch/utils/model/model_utils.py similarity index 100% rename from nlp/llm/Yi-1.5-6B/pytorch/utils/model/model_utils.py rename to nlp/llm/yi-1.5-6b/pytorch/utils/model/model_utils.py diff --git a/nlp/llm/Yi-1.5-6B/pytorch/utils/module/lora.py b/nlp/llm/yi-1.5-6b/pytorch/utils/module/lora.py similarity index 100% rename from nlp/llm/Yi-1.5-6B/pytorch/utils/module/lora.py rename to nlp/llm/yi-1.5-6b/pytorch/utils/module/lora.py diff --git a/nlp/llm/Yi-1.5-6B/pytorch/utils/perf.py b/nlp/llm/yi-1.5-6b/pytorch/utils/perf.py similarity index 100% rename from nlp/llm/Yi-1.5-6B/pytorch/utils/perf.py rename to nlp/llm/yi-1.5-6b/pytorch/utils/perf.py diff --git a/nlp/llm/Yi-1.5-6B/pytorch/utils/utils.py b/nlp/llm/yi-1.5-6b/pytorch/utils/utils.py similarity index 100% rename from nlp/llm/Yi-1.5-6B/pytorch/utils/utils.py rename to nlp/llm/yi-1.5-6b/pytorch/utils/utils.py diff --git a/nlp/llm/Yi-1.5-6B/pytorch/yi_example_dataset/data/eval.jsonl b/nlp/llm/yi-1.5-6b/pytorch/yi_example_dataset/data/eval.jsonl similarity index 100% rename from nlp/llm/Yi-1.5-6B/pytorch/yi_example_dataset/data/eval.jsonl rename to nlp/llm/yi-1.5-6b/pytorch/yi_example_dataset/data/eval.jsonl diff --git a/nlp/llm/Yi-1.5-6B/pytorch/yi_example_dataset/data/train.jsonl b/nlp/llm/yi-1.5-6b/pytorch/yi_example_dataset/data/train.jsonl similarity index 100% rename from nlp/llm/Yi-1.5-6B/pytorch/yi_example_dataset/data/train.jsonl rename to nlp/llm/yi-1.5-6b/pytorch/yi_example_dataset/data/train.jsonl diff --git a/nlp/llm/Yi-6B/pytorch/README.md b/nlp/llm/yi-6b/pytorch/README.md similarity index 100% rename from nlp/llm/Yi-6B/pytorch/README.md rename to nlp/llm/yi-6b/pytorch/README.md diff --git a/nlp/llm/Yi-6B/pytorch/constant.py b/nlp/llm/yi-6b/pytorch/constant.py similarity index 100% rename from nlp/llm/Yi-6B/pytorch/constant.py rename to nlp/llm/yi-6b/pytorch/constant.py diff --git a/nlp/llm/Yi-6B/pytorch/requirements.txt b/nlp/llm/yi-6b/pytorch/requirements.txt similarity index 100% rename from nlp/llm/Yi-6B/pytorch/requirements.txt rename to nlp/llm/yi-6b/pytorch/requirements.txt diff --git a/nlp/llm/Yi-6B/pytorch/scripts/run_sft_Yi_6b.sh b/nlp/llm/yi-6b/pytorch/scripts/run_sft_Yi_6b.sh similarity index 100% rename from nlp/llm/Yi-6B/pytorch/scripts/run_sft_Yi_6b.sh rename to nlp/llm/yi-6b/pytorch/scripts/run_sft_Yi_6b.sh diff --git a/nlp/llm/Yi-6B/pytorch/sft/main.py b/nlp/llm/yi-6b/pytorch/sft/main.py similarity index 100% rename from nlp/llm/Yi-6B/pytorch/sft/main.py rename to nlp/llm/yi-6b/pytorch/sft/main.py diff --git a/nlp/llm/Yi-6B/pytorch/sft/prompt_eval.py b/nlp/llm/yi-6b/pytorch/sft/prompt_eval.py similarity index 100% rename from nlp/llm/Yi-6B/pytorch/sft/prompt_eval.py rename to nlp/llm/yi-6b/pytorch/sft/prompt_eval.py diff --git a/nlp/llm/Yi-6B/pytorch/utils/data/data_utils.py b/nlp/llm/yi-6b/pytorch/utils/data/data_utils.py similarity index 100% rename from nlp/llm/Yi-6B/pytorch/utils/data/data_utils.py rename to nlp/llm/yi-6b/pytorch/utils/data/data_utils.py diff --git a/nlp/llm/Yi-6B/pytorch/utils/data/raw_datasets.py b/nlp/llm/yi-6b/pytorch/utils/data/raw_datasets.py similarity index 100% rename from nlp/llm/Yi-6B/pytorch/utils/data/raw_datasets.py rename to nlp/llm/yi-6b/pytorch/utils/data/raw_datasets.py diff --git a/nlp/llm/Yi-6B/pytorch/utils/ds_utils.py b/nlp/llm/yi-6b/pytorch/utils/ds_utils.py similarity index 100% rename from nlp/llm/Yi-6B/pytorch/utils/ds_utils.py rename to nlp/llm/yi-6b/pytorch/utils/ds_utils.py diff --git a/nlp/llm/Yi-6B/pytorch/utils/model/model_utils.py b/nlp/llm/yi-6b/pytorch/utils/model/model_utils.py similarity index 100% rename from nlp/llm/Yi-6B/pytorch/utils/model/model_utils.py rename to nlp/llm/yi-6b/pytorch/utils/model/model_utils.py diff --git a/nlp/llm/Yi-6B/pytorch/utils/module/lora.py b/nlp/llm/yi-6b/pytorch/utils/module/lora.py similarity index 100% rename from nlp/llm/Yi-6B/pytorch/utils/module/lora.py rename to nlp/llm/yi-6b/pytorch/utils/module/lora.py diff --git a/nlp/llm/Yi-6B/pytorch/utils/perf.py b/nlp/llm/yi-6b/pytorch/utils/perf.py similarity index 100% rename from nlp/llm/Yi-6B/pytorch/utils/perf.py rename to nlp/llm/yi-6b/pytorch/utils/perf.py diff --git a/nlp/llm/Yi-6B/pytorch/utils/utils.py b/nlp/llm/yi-6b/pytorch/utils/utils.py similarity index 100% rename from nlp/llm/Yi-6B/pytorch/utils/utils.py rename to nlp/llm/yi-6b/pytorch/utils/utils.py diff --git a/nlp/llm/Yi-6B/pytorch/yi_example_dataset/data/eval.jsonl b/nlp/llm/yi-6b/pytorch/yi_example_dataset/data/eval.jsonl similarity index 100% rename from nlp/llm/Yi-6B/pytorch/yi_example_dataset/data/eval.jsonl rename to nlp/llm/yi-6b/pytorch/yi_example_dataset/data/eval.jsonl diff --git a/nlp/llm/Yi-6B/pytorch/yi_example_dataset/data/train.jsonl b/nlp/llm/yi-6b/pytorch/yi_example_dataset/data/train.jsonl similarity index 100% rename from nlp/llm/Yi-6B/pytorch/yi_example_dataset/data/train.jsonl rename to nlp/llm/yi-6b/pytorch/yi_example_dataset/data/train.jsonl diff --git a/nlp/llm/Yi-VL-6B/pytorch/README.md b/nlp/llm/yi-vl-6b/pytorch/README.md similarity index 100% rename from nlp/llm/Yi-VL-6B/pytorch/README.md rename to nlp/llm/yi-vl-6b/pytorch/README.md diff --git a/nlp/llm/Yi-VL-6B/pytorch/yi_vl_6b_full_sft.yaml b/nlp/llm/yi-vl-6b/pytorch/yi_vl_6b_full_sft.yaml similarity index 100% rename from nlp/llm/Yi-VL-6B/pytorch/yi_vl_6b_full_sft.yaml rename to nlp/llm/yi-vl-6b/pytorch/yi_vl_6b_full_sft.yaml diff --git a/nlp/llm/Yi-VL-6B/pytorch/yi_vl_6b_lora_sft.yaml b/nlp/llm/yi-vl-6b/pytorch/yi_vl_6b_lora_sft.yaml similarity index 100% rename from nlp/llm/Yi-VL-6B/pytorch/yi_vl_6b_lora_sft.yaml rename to nlp/llm/yi-vl-6b/pytorch/yi_vl_6b_lora_sft.yaml