mushenL f77237b049 add llama2 pipeline (#399) ...

* Modify the parameter passing of the text_generation_pipeline class

* add llama2 pipeline

* add llama pipeline v1.1

* add llama pipeline v1.2

* add llama pipeline v1.3

* add llama pipeline v1.0.4

2023-07-22 21:53:04 +08:00

..

cli

fix torch 2.x compatible issue

2023-06-27 14:40:51 +08:00

export

Stable Diffusion model checkpoint export to onnx. (#340)

2023-06-28 13:26:19 +08:00

fileio

fix jsonplus, support ndarray

2023-03-21 18:10:10 +08:00

hub

Support FlexTrain and update the structure of trainer

2023-05-13 12:12:04 +08:00

metrics

Add trainer for UniTE

2023-05-11 14:41:08 +08:00

models

Refactor hooks

2023-02-28 13:51:01 +08:00

msdatasets

Support jsonl format in meta data

2023-06-27 11:58:19 +08:00

outputs

[to #42322933]fix bugs relate to token cls

2022-10-31 20:52:27 +08:00

pipelines

add llama2 pipeline (#399)

2023-07-22 21:53:04 +08:00

preprocessors

[to #47563396]Fix bug: two ckpt hooks save in the same dir

2023-02-13 16:54:02 +00:00

taskdataset

Refactor the task_datasets module

2023-03-10 09:03:32 +08:00

trainers

Update diffusers version to 0.18.0 (#377)

2023-07-14 19:02:52 +08:00

tuners

Add Lora/Adapter/Prompt and support for chatglm6B and chatglm2-6B

2023-06-27 14:38:18 +08:00

utils

Add Lora/Adapter/Prompt and support for chatglm6B and chatglm2-6B

2023-06-27 14:38:18 +08:00

__init__.py

[to #41402703] add basic modules

2022-05-17 10:15:00 +08:00

run_analysis.py

Refactor the task_datasets module

2023-03-10 09:03:32 +08:00

run_config.yaml

docker file py38 and py37 compatible merge

2023-06-08 10:01:58 +08:00

run.py

remove profiling log (#113)

2023-02-17 12:01:54 +08:00