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