AlpinDale fa84f8102e kernels: split marlin kernels for faster compile, fix MoE, temporarily remove HQQ (#1119) 1 개월 전
..
__init__.py a5bfc2bc3d VLM: add support for LLaVA-Onevision model (#1100) 1 개월 전
arctic.py 0dfa6b60ec core: support logprobs with multi-step scheduling (#963) 2 달 전
baichuan.py 0dfa6b60ec core: support logprobs with multi-step scheduling (#963) 2 달 전
bart.py a985143768 core: add cuda graph support for encoder-decoder models (#1051) 2 달 전
blip.py 7ce3174039 VLM: refactor blip models to support composite weight loading (#1098) 1 개월 전
blip2.py 7ce3174039 VLM: refactor blip models to support composite weight loading (#1098) 1 개월 전
bloom.py 0dfa6b60ec core: support logprobs with multi-step scheduling (#963) 2 달 전
chameleon.py 651678d2df VLM: use `SequenceData.from_token_counts` to create dummy data (#1093) 1 개월 전
chatglm.py 0dfa6b60ec core: support logprobs with multi-step scheduling (#963) 2 달 전
clip.py 80be38ca6f chore: expose phi3_v num_crops as an mm_processor_kwargs (#1117) 1 개월 전
commandr.py 135dfd648b fix: LoRA support for Cohere and Jamba models (#1004) 2 달 전
dbrx.py b65449b5ad moe: refactor DBRX experts to support FusedMoE (#1095) 1 개월 전
decilm.py 9022c6d869 remove progress_bar imports 4 달 전
deepseek.py 0dfa6b60ec core: support logprobs with multi-step scheduling (#963) 2 달 전
deepseek_v2.py 0dfa6b60ec core: support logprobs with multi-step scheduling (#963) 2 달 전
eagle.py 0dfa6b60ec core: support logprobs with multi-step scheduling (#963) 2 달 전
exaone.py 0dfa6b60ec core: support logprobs with multi-step scheduling (#963) 2 달 전
falcon.py 0dfa6b60ec core: support logprobs with multi-step scheduling (#963) 2 달 전
fuyu.py 0dfa6b60ec core: support logprobs with multi-step scheduling (#963) 2 달 전
gemma.py 0dfa6b60ec core: support logprobs with multi-step scheduling (#963) 2 달 전
gemma2.py b33cf04386 quants: add bitsandbytes support for gemma2 model (#1026) 2 달 전
gpt2.py 0dfa6b60ec core: support logprobs with multi-step scheduling (#963) 2 달 전
gpt_bigcode.py 0dfa6b60ec core: support logprobs with multi-step scheduling (#963) 2 달 전
gpt_j.py 0dfa6b60ec core: support logprobs with multi-step scheduling (#963) 2 달 전
gpt_neox.py 0dfa6b60ec core: support logprobs with multi-step scheduling (#963) 2 달 전
granite.py 11f49b5341 fix: granite logit scale in logit computation (#1054) 2 달 전
idefics2_vision_model.py f1d0b77c92 [0.6.0] Release Candidate (#481) 6 달 전
interfaces.py 0b8b407b6d feat: support profiling with multiple multi-modal inputs per prompt (#712) 6 달 전
intern_vit.py f56d6b396a vlm: fallback to SDPA for ViT models on CPU backend (#982) 2 달 전
internlm2.py 7632f91429 fix: InternLM2 model with Tensor Parallel (#980) 2 달 전
internvl.py 91d03c04d2 VLM: refactor composite weight loading logic (#1097) 1 개월 전
jais.py 0dfa6b60ec core: support logprobs with multi-step scheduling (#963) 2 달 전
jamba.py 135dfd648b fix: LoRA support for Cohere and Jamba models (#1004) 2 달 전
llama.py 0dfa6b60ec core: support logprobs with multi-step scheduling (#963) 2 달 전
llama_embedding.py 9022c6d869 remove progress_bar imports 4 달 전
llava.py 91d03c04d2 VLM: refactor composite weight loading logic (#1097) 1 개월 전
llava_next.py 7ce3174039 VLM: refactor blip models to support composite weight loading (#1098) 1 개월 전
llava_next_video.py 91d03c04d2 VLM: refactor composite weight loading logic (#1097) 1 개월 전
llava_onevision.py a5bfc2bc3d VLM: add support for LLaVA-Onevision model (#1100) 1 개월 전
mamba.py 3bb0f07461 chore: rename `task_handler` to `worker` (#985) 2 달 전
mamba_cache.py a113309876 kernel: add meta functions for ops to prevent graph breaks (#1019) 2 달 전
medusa.py 0dfa6b60ec core: support logprobs with multi-step scheduling (#963) 2 달 전
minicpm.py ce7b602f03 model: add support for MiniCPM-3 (#1044) 2 달 전
minicpm3.py ce7b602f03 model: add support for MiniCPM-3 (#1044) 2 달 전
minicpmv.py 651678d2df VLM: use `SequenceData.from_token_counts` to create dummy data (#1093) 1 개월 전
mixtral.py 0dfa6b60ec core: support logprobs with multi-step scheduling (#963) 2 달 전
mixtral_quant.py fa84f8102e kernels: split marlin kernels for faster compile, fix MoE, temporarily remove HQQ (#1119) 1 개월 전
mlp_speculator.py 0dfa6b60ec core: support logprobs with multi-step scheduling (#963) 2 달 전
molmo.py acc0c727c8 vlm: add support for molmo vision model (#1069) 2 달 전
mpt.py 0dfa6b60ec core: support logprobs with multi-step scheduling (#963) 2 달 전
na_vit.py 9f3e7c86e2 feat: add fused Marlin MoE kernel (#934) 2 달 전
nemotron.py 0dfa6b60ec core: support logprobs with multi-step scheduling (#963) 2 달 전
olmo.py 0dfa6b60ec core: support logprobs with multi-step scheduling (#963) 2 달 전
olmoe.py 0dfa6b60ec core: support logprobs with multi-step scheduling (#963) 2 달 전
opt.py 0dfa6b60ec core: support logprobs with multi-step scheduling (#963) 2 달 전
orion.py 0dfa6b60ec core: support logprobs with multi-step scheduling (#963) 2 달 전
paligemma.py 91d03c04d2 VLM: refactor composite weight loading logic (#1097) 1 개월 전
persimmon.py 0dfa6b60ec core: support logprobs with multi-step scheduling (#963) 2 달 전
phi.py 0dfa6b60ec core: support logprobs with multi-step scheduling (#963) 2 달 전
phi3.py ec17b6c4d0 fix: Phi3.5 Mini and MoE LoRA inference (#1070) 2 달 전
phi3_small.py 0dfa6b60ec core: support logprobs with multi-step scheduling (#963) 2 달 전
phi3v.py 80be38ca6f chore: expose phi3_v num_crops as an mm_processor_kwargs (#1117) 1 개월 전
phimoe.py ec17b6c4d0 fix: Phi3.5 Mini and MoE LoRA inference (#1070) 2 달 전
pixtral.py 651678d2df VLM: use `SequenceData.from_token_counts` to create dummy data (#1093) 1 개월 전
qwen.py 651678d2df VLM: use `SequenceData.from_token_counts` to create dummy data (#1093) 1 개월 전
qwen2.py c90abcc603 VLM: add pipeline parallelism support for Qwen2-VL (#1103) 1 개월 전
qwen2_moe.py 5224389dae chore: skip loading extra bias for qwen2 moe GPTQ (#1011) 2 달 전
qwen2_vl.py c90abcc603 VLM: add pipeline parallelism support for Qwen2-VL (#1103) 1 개월 전
siglip.py a5bfc2bc3d VLM: add support for LLaVA-Onevision model (#1100) 1 개월 전
solar.py 0dfa6b60ec core: support logprobs with multi-step scheduling (#963) 2 달 전
stablelm.py 0dfa6b60ec core: support logprobs with multi-step scheduling (#963) 2 달 전
starcoder2.py 0dfa6b60ec core: support logprobs with multi-step scheduling (#963) 2 달 전
ultravox.py 18e0b0e932 chore: support loading weights by ID within models (#1116) 1 개월 전
utils.py 91d03c04d2 VLM: refactor composite weight loading logic (#1097) 1 개월 전
xverse.py 0dfa6b60ec core: support logprobs with multi-step scheduling (#963) 2 달 전