| .. |
|
__init__.py
|
0256ed236b
feat: windows support (#790)
|
2 months ago |
|
arctic.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
baichuan.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
bart.py
|
9576096b9d
iterate over weights normally
|
2 months ago |
|
blip.py
|
2f61644f6e
SPMD optimizations (#824)
|
1 month ago |
|
blip2.py
|
2f61644f6e
SPMD optimizations (#824)
|
1 month ago |
|
bloom.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
chameleon.py
|
2f61644f6e
SPMD optimizations (#824)
|
1 month ago |
|
chatglm.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
clip.py
|
2f61644f6e
SPMD optimizations (#824)
|
1 month ago |
|
commandr.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
dbrx.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
decilm.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
deepseek.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
deepseek_v2.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
exaone.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
falcon.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
fuyu.py
|
2f61644f6e
SPMD optimizations (#824)
|
1 month ago |
|
gemma.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
gemma2.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
gpt2.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
gpt_bigcode.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
gpt_j.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
gpt_neox.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
idefics2_vision_model.py
|
f1d0b77c92
[0.6.0] Release Candidate (#481)
|
4 months ago |
|
interfaces.py
|
0b8b407b6d
feat: support profiling with multiple multi-modal inputs per prompt (#712)
|
3 months ago |
|
intern_vit.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
internlm2.py
|
e182d00256
feat: AWQ quantization for InternVL (#867)
|
1 month ago |
|
internvl.py
|
0b8b407b6d
feat: support profiling with multiple multi-modal inputs per prompt (#712)
|
3 months ago |
|
jais.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
jamba.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
llama.py
|
1a5e612eaf
pass tensor progress bar as weight iterator
|
2 months ago |
|
llama_embedding.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
llava.py
|
0b8b407b6d
feat: support profiling with multiple multi-modal inputs per prompt (#712)
|
3 months ago |
|
llava_next.py
|
0b8b407b6d
feat: support profiling with multiple multi-modal inputs per prompt (#712)
|
3 months ago |
|
mamba.py
|
0e558e9b2f
fix: loading chameleon model with TP>1 (#695)
|
4 months ago |
|
mamba_cache.py
|
bf88c8567e
feat: mamba model support (#674)
|
4 months ago |
|
medusa.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
minicpm.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
minicpmv.py
|
2f61644f6e
SPMD optimizations (#824)
|
1 month ago |
|
mixtral.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
mixtral_quant.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
mlp_speculator.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
mpt.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
na_vit.py
|
f1d0b77c92
[0.6.0] Release Candidate (#481)
|
4 months ago |
|
nemotron.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
olmo.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
olmoe.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
opt.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
orion.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
paligemma.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
persimmon.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
phi.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
phi3_small.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
phi3v.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
qwen.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
qwen2.py
|
1a5e612eaf
pass tensor progress bar as weight iterator
|
2 months ago |
|
qwen2_moe.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
siglip.py
|
2f61644f6e
SPMD optimizations (#824)
|
1 month ago |
|
solar.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
stablelm.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
starcoder2.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
|
utils.py
|
3693028340
feat: support for Audio modality (#698)
|
4 months ago |
|
xverse.py
|
9022c6d869
remove progress_bar imports
|
2 months ago |
AlpinDale
e182d00256
feat: AWQ quantization for InternVL (#867)