Making large AI models cheaper, faster and more accessible

aibig-modeldata-parallelismdeep-learningdistributed-computingfoundation-modelsheterogeneous-traininghpcinferencelarge-scalemodel-parallelismpipeline-parallelism

You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

Tree: 0b81163bc0

ColossalChat

ckpt

cloud/coati

colossalchat

colossalchat_upgrade

dev/gpt2_metainfo_patch

dev/zero-offload

develop

feat/moe

feat/online-serving

feat/speculative-decoding

feature/2-stage

feature/async-io

feature/colossal-infer

feature/elixir

feature/inference-refactor

feature/lora

feature/stable-diffusion

feature/zerobubble

fix-setup

fix/format

flybird11111-patch-1

hotfix/example_test

hotfix/kernel_build_before_load

llama3

main

moe_sp

pre-commit-ci-update-config

refactor/inference

supercooledith-patch-1

support-npu

v0.0.1-beta

v0.0.2

v0.1.0

v0.1.1

v0.1.10

v0.1.11rc1

v0.1.11rc2

v0.1.11rc3

v0.1.11rc4

v0.1.11rc5

v0.1.12

v0.1.13

v0.1.2

v0.1.3

v0.1.4

v0.1.5

v0.1.6

v0.1.7

v0.1.8

v0.1.9

v0.2.0

v0.2.1

v0.2.2

v0.2.3

v0.2.4

v0.2.5

v0.2.6

v0.2.7

v0.2.8

v0.3.0

v0.3.1

v0.3.2

v0.3.3

v0.3.4

v0.3.5

v0.3.6

v0.3.7

v0.3.8

v0.3.9

v0.4.0

v0.4.1

v0.4.2

v0.4.3

v0.4.4

v0.4.5

v0.4.6

Branches Tags

${ item.name }

Create tag ${ searchTerm }

Create branch ${ searchTerm }

from '0b81163bc0'

${ noResults }

ColossalAI/tests/test_infer

History

flybird11111 2ddf624a86 [shardformer] upgrade transformers to 4.39.3 (#5815) ... * [shardformer]upgrade transformers for gpt2/gptj/whisper (#5807) * [shardformer] fix modeling of gpt2 and gptj * [shardformer] fix whisper modeling * [misc] update requirements --------- Co-authored-by: ver217 <lhx0217@gmail.com> * [shardformer]upgrade transformers for mistral (#5808) * upgrade transformers for mistral * fix * fix * [shardformer]upgrade transformers for llama (#5809) * update transformers fix * fix * fix * [inference] upgrade transformers (#5810) * update transformers fix * fix * fix * fix * fix * [gemini] update transformers for gemini (#5814) --------- Co-authored-by: ver217 <lhx0217@gmail.com>		5 months ago
..
test_async_engine	[Inference] Fix bugs and docs for feat/online-server (#5598)	7 months ago
test_kernels	[shardformer] upgrade transformers to 4.39.3 (#5815)	5 months ago
test_models	[Inference] Fix flash-attn import and add model test (#5794)	5 months ago
__init__.py	[Fix] Fix Inference Example, Tests, and Requirements (#5688)	7 months ago
_utils.py	[Inference] Add the logic of the inference engine (#5173)	11 months ago
test_batch_bucket.py	[Fix/Inference] Fix format of input prompts and input model in inference engine (#5395)	9 months ago
test_config_and_struct.py	[Fix] Fix Inference Example, Tests, and Requirements (#5688)	7 months ago
test_continuous_batching.py	[inference] Fix running time of test_continuous_batching (#5750)	6 months ago
test_cuda_graph.py	[Fix] Fix Inference Example, Tests, and Requirements (#5688)	7 months ago
test_drafter.py	[Fix] Fix Inference Example, Tests, and Requirements (#5688)	7 months ago
test_inference_engine.py	[Inference] Fix bugs and docs for feat/online-server (#5598)	7 months ago
test_kvcache_manager.py	[Fix] Fix & Update Inference Tests (compatibility w/ main)	7 months ago
test_request_handler.py	[Fix] Fix & Update Inference Tests (compatibility w/ main)	7 months ago
test_rpc_engine.py	[release] update version (#5752)	6 months ago
test_streamingllm.py	[Inference]Add Streaming LLM (#5745)	6 months ago