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: 5f8c0a0ac3

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 '5f8c0a0ac3'

${ noResults }

ColossalAI/colossalai/shardformer/modeling

History

Haze188 22ce873c3f [Shardformer] Add parallel output for shardformer models(bloom, falcon) (#5702) ... * [pre-commit.ci] auto fixes from pre-commit.com hooks * add parallel cross entropy output for falcon model & fix some typos in bloom.py * fix module name error, self.model -> self.transformers in bloom, falcon model * Fix the overflow bug of distributed cross entropy loss function when training with fp16 * add dtype to parallel cross entropy loss function * fix dtype related typos adn prettify the loss.py * fix grad dtype and update dtype mismatch error * fix typo bugs		6 months ago
..
chatglm2_6b	[hotfix] fix typo change enabel to enable under colossalai/shardformer/ (#5317)	9 months ago
__init__.py	[shardformer] added development protocol for standardization (#4149)	1 year ago
bert.py	[shardformer] support bias_gelu_jit_fused for models (#5647)	7 months ago
blip2.py	[shardformer] support bias_gelu_jit_fused for models (#5647)	7 months ago
bloom.py	[Shardformer] Add parallel output for shardformer models(bloom, falcon) (#5702)	6 months ago
chatglm2.py	[shardformer] fix chatglm implementation (#5644)	7 months ago
falcon.py	[Shardformer] Add parallel output for shardformer models(bloom, falcon) (#5702)	6 months ago
gpt2.py	[Shardformer] Add parallel output for shardformer models(bloom, falcon) (#5702)	6 months ago
gptj.py	[shardformer] update transformers (#5583)	7 months ago
jit.py	[misc] update pre-commit and run all files (#4752)	1 year ago
llama.py	[Shardformer] Add parallel output for shardformer models(bloom, falcon) (#5702)	6 months ago
mistral.py	[Shardformer] Add parallel output for shardformer models(bloom, falcon) (#5702)	6 months ago
opt.py	[Shardformer] Add parallel output for shardformer models(bloom, falcon) (#5702)	6 months ago
qwen2.py	[Shardformer]fix the num_heads assert for llama model and qwen model (#5704)	7 months ago
sam.py	[misc] update pre-commit and run all files (#4752)	1 year ago
t5.py	[shardformer] update transformers (#5583)	7 months ago
vit.py	[shardformer] support bias_gelu_jit_fused for models (#5647)	7 months ago
whisper.py	[shardformer] update transformers (#5583)	7 months ago