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