Lianmin Zheng be0124bda0 Rename triton_fused_moe -> fused_moe_triton (#2163)

2024-11-24 08:12:35 -08:00

..

baichuan.py

Update vocab embedding deps and add TP switch (#1856)

2024-10-31 20:13:07 -07:00

chatglm.py

docs: fix module docstrings and copyright headers (#2077)

2024-11-22 22:16:53 +08:00

commandr.py

docs: fix module docstrings and copyright headers (#2077)

2024-11-22 22:16:53 +08:00

dbrx.py

Rename triton_fused_moe -> fused_moe_triton (#2163)

2024-11-24 08:12:35 -08:00

deepseek_v2.py

Rename triton_fused_moe -> fused_moe_triton (#2163)

2024-11-24 08:12:35 -08:00

deepseek.py

Rename triton_fused_moe -> fused_moe_triton (#2163)

2024-11-24 08:12:35 -08:00

exaone.py

docs: fix module docstrings and copyright headers (#2077)

2024-11-22 22:16:53 +08:00

gemma2_reward.py

docs: fix module docstrings and copyright headers (#2077)

2024-11-22 22:16:53 +08:00

gemma2.py

Add simple CPU offloading support. (#2081)

2024-11-23 06:23:53 +00:00

gemma.py

docs: fix module docstrings and copyright headers (#2077)

2024-11-22 22:16:53 +08:00

gpt2.py

fix black in pre-commit (#1940)

2024-11-08 07:42:47 +08:00

gpt_bigcode.py

docs: fix module docstrings and copyright headers (#2077)

2024-11-22 22:16:53 +08:00

grok.py

Rename triton_fused_moe -> fused_moe_triton (#2163)

2024-11-24 08:12:35 -08:00

internlm2_reward.py

docs: fix module docstrings and copyright headers (#2077)

2024-11-22 22:16:53 +08:00

internlm2.py

docs: fix module docstrings and copyright headers (#2077)

2024-11-22 22:16:53 +08:00

llama_classification.py

docs: fix module docstrings and copyright headers (#2077)

2024-11-22 22:16:53 +08:00

llama_embedding.py

Let reward model take text inputs instead of message lists (#1907)

2024-11-03 13:27:12 -08:00

llama_reward.py

docs: fix module docstrings and copyright headers (#2077)

2024-11-22 22:16:53 +08:00

llama.py

Add simple CPU offloading support. (#2081)

2024-11-23 06:23:53 +00:00

llava.py

docs: fix module docstrings and copyright headers (#2077)

2024-11-22 22:16:53 +08:00

llavavid.py

docs: fix module docstrings and copyright headers (#2077)

2024-11-22 22:16:53 +08:00

minicpm3.py

docs: fix module docstrings and copyright headers (#2077)

2024-11-22 22:16:53 +08:00

minicpm.py

docs: fix module docstrings and copyright headers (#2077)

2024-11-22 22:16:53 +08:00

mistral.py

docs: fix module docstrings and copyright headers (#2077)

2024-11-22 22:16:53 +08:00

mixtral_quant.py

docs: fix module docstrings and copyright headers (#2077)

2024-11-22 22:16:53 +08:00

mixtral.py

Rename triton_fused_moe -> fused_moe_triton (#2163)

2024-11-24 08:12:35 -08:00

mllama.py

Update vocab embedding deps and add TP switch (#1856)

2024-10-31 20:13:07 -07:00

olmo.py

Add simple CPU offloading support. (#2081)

2024-11-23 06:23:53 +00:00

olmoe.py

Rename triton_fused_moe -> fused_moe_triton (#2163)

2024-11-24 08:12:35 -08:00

phi3_small.py

Crash the CI jobs on model import errors (#2072)

2024-11-17 22:18:11 -08:00

qwen2_moe.py

Rename triton_fused_moe -> fused_moe_triton (#2163)

2024-11-24 08:12:35 -08:00

qwen2_vl.py

Add support for Qwen2-VL-based embedding models (#2055)

2024-11-21 14:24:25 -08:00

qwen2.py

Add simple CPU offloading support. (#2081)

2024-11-23 06:23:53 +00:00

qwen.py

docs: fix module docstrings and copyright headers (#2077)

2024-11-22 22:16:53 +08:00

stablelm.py

docs: fix module docstrings and copyright headers (#2077)

2024-11-22 22:16:53 +08:00

torch_native_llama.py

docs: fix module docstrings and copyright headers (#2077)

2024-11-22 22:16:53 +08:00

xverse_moe.py

Rename triton_fused_moe -> fused_moe_triton (#2163)

2024-11-24 08:12:35 -08:00

xverse.py

docs: fix module docstrings and copyright headers (#2077)

2024-11-22 22:16:53 +08:00

yivl.py

docs: fix module docstrings and copyright headers (#2077)

2024-11-22 22:16:53 +08:00