ColossalAI

Commit Graph

Author	SHA1	Message	Date
Yuanheng Zhao	7b249c76e5	[Fix] Fix spec-dec Glide LlamaModel for compatibility with transformers (#5837 ) * fix glide llama model * revise	2024-06-19 15:37:53 +08:00
Yuanheng Zhao	d85d91435a	[Inference/SpecDec] Support GLIDE Drafter Model (#5455 ) * add glide-llama policy and modeling * update glide modeling, compitable with transformers 4.36.2 * revise glide llama modeling/usage * fix issues of glimpsing large kv * revise the way re-loading params for glide drafter * fix drafter and engine tests * enable convert to glide strict=False * revise glide llama modeling * revise vicuna prompt template * revise drafter and tests * apply usage of glide model in engine	2024-04-10 11:07:52 +08:00
Yuanheng Zhao	5a9b05f7b2	[Inference/SpecDec] Add Basic Drafter Model Container (#5405 ) * [Infer/Fix] Fix Dependency in test - RMSNorm kernel (#5399) fix dependency in pytest * add drafter model container (basic ver)	2024-04-10 11:07:51 +08:00

Author

SHA1

Message

Date

Yuanheng Zhao

7b249c76e5

[Fix] Fix spec-dec Glide LlamaModel for compatibility with transformers (#5837 )

* fix glide llama model

* revise

2024-06-19 15:37:53 +08:00

Yuanheng Zhao

d85d91435a

[Inference/SpecDec] Support GLIDE Drafter Model (#5455 )

* add glide-llama policy and modeling

* update glide modeling, compitable with transformers 4.36.2

* revise glide llama modeling/usage

* fix issues of glimpsing large kv

* revise the way re-loading params for glide drafter

* fix drafter and engine tests

* enable convert to glide strict=False

* revise glide llama modeling

* revise vicuna prompt template

* revise drafter and tests

* apply usage of glide model in engine

2024-04-10 11:07:52 +08:00

Yuanheng Zhao

5a9b05f7b2

[Inference/SpecDec] Add Basic Drafter Model Container (#5405 )

* [Infer/Fix] Fix Dependency in test - RMSNorm kernel (#5399)

fix dependency in pytest

* add drafter model container (basic ver)

2024-04-10 11:07:51 +08:00

3 Commits (23199e34cc0ae6fdd6c9297ef088bf4dc4713b52)