vllm-cpu 0.11.0.post2__cp312-cp312-manylinux_2_17_x86_64.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (1398) hide show
  1. vllm/_C.abi3.so +0 -0
  2. vllm/__init__.py +220 -0
  3. vllm/_bc_linter.py +59 -0
  4. vllm/_custom_ops.py +2044 -0
  5. vllm/_ipex_ops.py +393 -0
  6. vllm/_version.py +34 -0
  7. vllm/assets/__init__.py +0 -0
  8. vllm/assets/audio.py +45 -0
  9. vllm/assets/base.py +41 -0
  10. vllm/assets/image.py +50 -0
  11. vllm/assets/video.py +145 -0
  12. vllm/attention/__init__.py +15 -0
  13. vllm/attention/backends/__init__.py +0 -0
  14. vllm/attention/backends/abstract.py +204 -0
  15. vllm/attention/backends/utils.py +33 -0
  16. vllm/attention/layer.py +645 -0
  17. vllm/attention/layers/__init__.py +0 -0
  18. vllm/attention/layers/chunked_local_attention.py +93 -0
  19. vllm/attention/layers/cross_attention.py +162 -0
  20. vllm/attention/layers/encoder_only_attention.py +86 -0
  21. vllm/attention/ops/__init__.py +0 -0
  22. vllm/attention/ops/chunked_prefill_paged_decode.py +405 -0
  23. vllm/attention/ops/common.py +345 -0
  24. vllm/attention/ops/flashmla.py +192 -0
  25. vllm/attention/ops/merge_attn_states.py +43 -0
  26. vllm/attention/ops/paged_attn.py +262 -0
  27. vllm/attention/ops/pallas_kv_cache_update.py +124 -0
  28. vllm/attention/ops/prefix_prefill.py +928 -0
  29. vllm/attention/ops/rocm_aiter_mla.py +104 -0
  30. vllm/attention/ops/rocm_aiter_paged_attn.py +102 -0
  31. vllm/attention/ops/triton_decode_attention.py +691 -0
  32. vllm/attention/ops/triton_flash_attention.py +984 -0
  33. vllm/attention/ops/triton_merge_attn_states.py +97 -0
  34. vllm/attention/ops/triton_reshape_and_cache_flash.py +175 -0
  35. vllm/attention/ops/triton_unified_attention.py +894 -0
  36. vllm/attention/selector.py +245 -0
  37. vllm/attention/utils/__init__.py +0 -0
  38. vllm/attention/utils/fa_utils.py +85 -0
  39. vllm/attention/utils/kv_sharing_utils.py +33 -0
  40. vllm/beam_search.py +87 -0
  41. vllm/benchmarks/__init__.py +0 -0
  42. vllm/benchmarks/datasets.py +2723 -0
  43. vllm/benchmarks/latency.py +170 -0
  44. vllm/benchmarks/lib/__init__.py +3 -0
  45. vllm/benchmarks/lib/endpoint_request_func.py +533 -0
  46. vllm/benchmarks/lib/ready_checker.py +73 -0
  47. vllm/benchmarks/lib/utils.py +80 -0
  48. vllm/benchmarks/serve.py +1358 -0
  49. vllm/benchmarks/throughput.py +696 -0
  50. vllm/collect_env.py +823 -0
  51. vllm/compilation/__init__.py +0 -0
  52. vllm/compilation/activation_quant_fusion.py +189 -0
  53. vllm/compilation/backends.py +650 -0
  54. vllm/compilation/base_static_graph.py +56 -0
  55. vllm/compilation/collective_fusion.py +1188 -0
  56. vllm/compilation/compiler_interface.py +573 -0
  57. vllm/compilation/counter.py +47 -0
  58. vllm/compilation/cuda_graph.py +199 -0
  59. vllm/compilation/cuda_piecewise_backend.py +117 -0
  60. vllm/compilation/decorators.py +400 -0
  61. vllm/compilation/fix_functionalization.py +205 -0
  62. vllm/compilation/fusion.py +383 -0
  63. vllm/compilation/fusion_attn.py +295 -0
  64. vllm/compilation/fx_utils.py +84 -0
  65. vllm/compilation/inductor_pass.py +136 -0
  66. vllm/compilation/monitor.py +57 -0
  67. vllm/compilation/noop_elimination.py +158 -0
  68. vllm/compilation/pass_manager.py +125 -0
  69. vllm/compilation/post_cleanup.py +20 -0
  70. vllm/compilation/sequence_parallelism.py +478 -0
  71. vllm/compilation/torch25_custom_graph_pass.py +42 -0
  72. vllm/compilation/vllm_inductor_pass.py +156 -0
  73. vllm/compilation/wrapper.py +136 -0
  74. vllm/config/__init__.py +814 -0
  75. vllm/config/cache.py +220 -0
  76. vllm/config/compilation.py +673 -0
  77. vllm/config/device.py +74 -0
  78. vllm/config/kv_events.py +50 -0
  79. vllm/config/kv_transfer.py +111 -0
  80. vllm/config/load.py +113 -0
  81. vllm/config/lora.py +132 -0
  82. vllm/config/model.py +1912 -0
  83. vllm/config/multimodal.py +129 -0
  84. vllm/config/observability.py +99 -0
  85. vllm/config/parallel.py +524 -0
  86. vllm/config/pooler.py +97 -0
  87. vllm/config/scheduler.py +287 -0
  88. vllm/config/speculative.py +568 -0
  89. vllm/config/speech_to_text.py +39 -0
  90. vllm/config/structured_outputs.py +64 -0
  91. vllm/config/utils.py +145 -0
  92. vllm/connections.py +186 -0
  93. vllm/device_allocator/__init__.py +0 -0
  94. vllm/device_allocator/cumem.py +311 -0
  95. vllm/distributed/__init__.py +6 -0
  96. vllm/distributed/communication_op.py +41 -0
  97. vllm/distributed/device_communicators/__init__.py +0 -0
  98. vllm/distributed/device_communicators/all2all.py +440 -0
  99. vllm/distributed/device_communicators/all_reduce_utils.py +317 -0
  100. vllm/distributed/device_communicators/base_device_communicator.py +295 -0
  101. vllm/distributed/device_communicators/cpu_communicator.py +201 -0
  102. vllm/distributed/device_communicators/cuda_communicator.py +323 -0
  103. vllm/distributed/device_communicators/cuda_wrapper.py +180 -0
  104. vllm/distributed/device_communicators/custom_all_reduce.py +311 -0
  105. vllm/distributed/device_communicators/mnnvl_compat.py +28 -0
  106. vllm/distributed/device_communicators/pynccl.py +340 -0
  107. vllm/distributed/device_communicators/pynccl_allocator.py +186 -0
  108. vllm/distributed/device_communicators/pynccl_wrapper.py +416 -0
  109. vllm/distributed/device_communicators/quick_all_reduce.py +278 -0
  110. vllm/distributed/device_communicators/ray_communicator.py +258 -0
  111. vllm/distributed/device_communicators/shm_broadcast.py +589 -0
  112. vllm/distributed/device_communicators/shm_object_storage.py +635 -0
  113. vllm/distributed/device_communicators/symm_mem.py +136 -0
  114. vllm/distributed/device_communicators/tpu_communicator.py +102 -0
  115. vllm/distributed/device_communicators/xpu_communicator.py +94 -0
  116. vllm/distributed/eplb/__init__.py +8 -0
  117. vllm/distributed/eplb/eplb_state.py +620 -0
  118. vllm/distributed/eplb/rebalance_algo.py +239 -0
  119. vllm/distributed/eplb/rebalance_execute.py +424 -0
  120. vllm/distributed/kv_events.py +362 -0
  121. vllm/distributed/kv_transfer/README.md +29 -0
  122. vllm/distributed/kv_transfer/__init__.py +13 -0
  123. vllm/distributed/kv_transfer/disagg_prefill_workflow.jpg +0 -0
  124. vllm/distributed/kv_transfer/kv_connector/__init__.py +0 -0
  125. vllm/distributed/kv_transfer/kv_connector/base.py +10 -0
  126. vllm/distributed/kv_transfer/kv_connector/factory.py +113 -0
  127. vllm/distributed/kv_transfer/kv_connector/utils.py +261 -0
  128. vllm/distributed/kv_transfer/kv_connector/v1/__init__.py +6 -0
  129. vllm/distributed/kv_transfer/kv_connector/v1/base.py +388 -0
  130. vllm/distributed/kv_transfer/kv_connector/v1/lmcache_connector.py +168 -0
  131. vllm/distributed/kv_transfer/kv_connector/v1/metrics.py +100 -0
  132. vllm/distributed/kv_transfer/kv_connector/v1/multi_connector.py +328 -0
  133. vllm/distributed/kv_transfer/kv_connector/v1/nixl_connector.py +1473 -0
  134. vllm/distributed/kv_transfer/kv_connector/v1/offloading_connector.py +485 -0
  135. vllm/distributed/kv_transfer/kv_connector/v1/p2p/__init__.py +0 -0
  136. vllm/distributed/kv_transfer/kv_connector/v1/p2p/p2p_nccl_connector.py +488 -0
  137. vllm/distributed/kv_transfer/kv_connector/v1/p2p/p2p_nccl_engine.py +550 -0
  138. vllm/distributed/kv_transfer/kv_connector/v1/p2p/tensor_memory_pool.py +267 -0
  139. vllm/distributed/kv_transfer/kv_connector/v1/shared_storage_connector.py +418 -0
  140. vllm/distributed/kv_transfer/kv_lookup_buffer/__init__.py +0 -0
  141. vllm/distributed/kv_transfer/kv_lookup_buffer/base.py +175 -0
  142. vllm/distributed/kv_transfer/kv_lookup_buffer/mooncake_store.py +161 -0
  143. vllm/distributed/kv_transfer/kv_lookup_buffer/simple_buffer.py +237 -0
  144. vllm/distributed/kv_transfer/kv_pipe/__init__.py +0 -0
  145. vllm/distributed/kv_transfer/kv_pipe/base.py +67 -0
  146. vllm/distributed/kv_transfer/kv_pipe/mooncake_pipe.py +290 -0
  147. vllm/distributed/kv_transfer/kv_pipe/pynccl_pipe.py +280 -0
  148. vllm/distributed/kv_transfer/kv_transfer_state.py +73 -0
  149. vllm/distributed/parallel_state.py +1532 -0
  150. vllm/distributed/tpu_distributed_utils.py +178 -0
  151. vllm/distributed/utils.py +536 -0
  152. vllm/engine/__init__.py +0 -0
  153. vllm/engine/arg_utils.py +1778 -0
  154. vllm/engine/async_llm_engine.py +6 -0
  155. vllm/engine/llm_engine.py +6 -0
  156. vllm/engine/metrics.py +577 -0
  157. vllm/engine/metrics_types.py +84 -0
  158. vllm/engine/protocol.py +333 -0
  159. vllm/entrypoints/__init__.py +0 -0
  160. vllm/entrypoints/api_server.py +178 -0
  161. vllm/entrypoints/chat_utils.py +1705 -0
  162. vllm/entrypoints/cli/__init__.py +12 -0
  163. vllm/entrypoints/cli/benchmark/__init__.py +0 -0
  164. vllm/entrypoints/cli/benchmark/base.py +25 -0
  165. vllm/entrypoints/cli/benchmark/latency.py +21 -0
  166. vllm/entrypoints/cli/benchmark/main.py +55 -0
  167. vllm/entrypoints/cli/benchmark/serve.py +21 -0
  168. vllm/entrypoints/cli/benchmark/throughput.py +21 -0
  169. vllm/entrypoints/cli/collect_env.py +36 -0
  170. vllm/entrypoints/cli/main.py +60 -0
  171. vllm/entrypoints/cli/openai.py +233 -0
  172. vllm/entrypoints/cli/run_batch.py +67 -0
  173. vllm/entrypoints/cli/serve.py +232 -0
  174. vllm/entrypoints/cli/types.py +29 -0
  175. vllm/entrypoints/constants.py +10 -0
  176. vllm/entrypoints/context.py +481 -0
  177. vllm/entrypoints/harmony_utils.py +436 -0
  178. vllm/entrypoints/launcher.py +164 -0
  179. vllm/entrypoints/llm.py +1629 -0
  180. vllm/entrypoints/logger.py +79 -0
  181. vllm/entrypoints/openai/__init__.py +0 -0
  182. vllm/entrypoints/openai/api_server.py +1953 -0
  183. vllm/entrypoints/openai/cli_args.py +288 -0
  184. vllm/entrypoints/openai/logits_processors.py +90 -0
  185. vllm/entrypoints/openai/protocol.py +2757 -0
  186. vllm/entrypoints/openai/run_batch.py +491 -0
  187. vllm/entrypoints/openai/serving_chat.py +1597 -0
  188. vllm/entrypoints/openai/serving_classification.py +173 -0
  189. vllm/entrypoints/openai/serving_completion.py +692 -0
  190. vllm/entrypoints/openai/serving_embedding.py +631 -0
  191. vllm/entrypoints/openai/serving_engine.py +992 -0
  192. vllm/entrypoints/openai/serving_models.py +288 -0
  193. vllm/entrypoints/openai/serving_pooling.py +276 -0
  194. vllm/entrypoints/openai/serving_responses.py +1709 -0
  195. vllm/entrypoints/openai/serving_score.py +479 -0
  196. vllm/entrypoints/openai/serving_tokenization.py +196 -0
  197. vllm/entrypoints/openai/serving_transcription.py +136 -0
  198. vllm/entrypoints/openai/speech_to_text.py +388 -0
  199. vllm/entrypoints/openai/tool_parsers/__init__.py +55 -0
  200. vllm/entrypoints/openai/tool_parsers/abstract_tool_parser.py +164 -0
  201. vllm/entrypoints/openai/tool_parsers/deepseekv31_tool_parser.py +367 -0
  202. vllm/entrypoints/openai/tool_parsers/deepseekv3_tool_parser.py +370 -0
  203. vllm/entrypoints/openai/tool_parsers/glm4_moe_tool_parser.py +185 -0
  204. vllm/entrypoints/openai/tool_parsers/granite_20b_fc_tool_parser.py +259 -0
  205. vllm/entrypoints/openai/tool_parsers/granite_tool_parser.py +237 -0
  206. vllm/entrypoints/openai/tool_parsers/hermes_tool_parser.py +455 -0
  207. vllm/entrypoints/openai/tool_parsers/hunyuan_a13b_tool_parser.py +372 -0
  208. vllm/entrypoints/openai/tool_parsers/internlm2_tool_parser.py +216 -0
  209. vllm/entrypoints/openai/tool_parsers/jamba_tool_parser.py +308 -0
  210. vllm/entrypoints/openai/tool_parsers/kimi_k2_tool_parser.py +377 -0
  211. vllm/entrypoints/openai/tool_parsers/llama4_pythonic_tool_parser.py +316 -0
  212. vllm/entrypoints/openai/tool_parsers/llama_tool_parser.py +269 -0
  213. vllm/entrypoints/openai/tool_parsers/longcat_tool_parser.py +39 -0
  214. vllm/entrypoints/openai/tool_parsers/minimax_tool_parser.py +816 -0
  215. vllm/entrypoints/openai/tool_parsers/mistral_tool_parser.py +369 -0
  216. vllm/entrypoints/openai/tool_parsers/openai_tool_parser.py +93 -0
  217. vllm/entrypoints/openai/tool_parsers/phi4mini_tool_parser.py +112 -0
  218. vllm/entrypoints/openai/tool_parsers/pythonic_tool_parser.py +308 -0
  219. vllm/entrypoints/openai/tool_parsers/qwen3coder_tool_parser.py +707 -0
  220. vllm/entrypoints/openai/tool_parsers/qwen3xml_tool_parser.py +1137 -0
  221. vllm/entrypoints/openai/tool_parsers/seed_oss_tool_parser.py +679 -0
  222. vllm/entrypoints/openai/tool_parsers/step3_tool_parser.py +296 -0
  223. vllm/entrypoints/openai/tool_parsers/utils.py +124 -0
  224. vllm/entrypoints/openai/tool_parsers/xlam_tool_parser.py +524 -0
  225. vllm/entrypoints/renderer.py +395 -0
  226. vllm/entrypoints/score_utils.py +232 -0
  227. vllm/entrypoints/ssl.py +75 -0
  228. vllm/entrypoints/tool.py +139 -0
  229. vllm/entrypoints/tool_server.py +206 -0
  230. vllm/entrypoints/utils.py +233 -0
  231. vllm/env_override.py +23 -0
  232. vllm/envs.py +1590 -0
  233. vllm/executor/__init__.py +0 -0
  234. vllm/executor/executor_base.py +381 -0
  235. vllm/executor/msgspec_utils.py +35 -0
  236. vllm/executor/ray_distributed_executor.py +699 -0
  237. vllm/executor/ray_utils.py +410 -0
  238. vllm/executor/uniproc_executor.py +176 -0
  239. vllm/forward_context.py +402 -0
  240. vllm/inputs/__init__.py +30 -0
  241. vllm/inputs/data.py +356 -0
  242. vllm/inputs/parse.py +151 -0
  243. vllm/inputs/preprocess.py +664 -0
  244. vllm/logger.py +229 -0
  245. vllm/logging_utils/__init__.py +10 -0
  246. vllm/logging_utils/dump_input.py +81 -0
  247. vllm/logging_utils/formatter.py +79 -0
  248. vllm/logging_utils/log_time.py +32 -0
  249. vllm/logits_process.py +119 -0
  250. vllm/logprobs.py +28 -0
  251. vllm/lora/__init__.py +0 -0
  252. vllm/lora/layers/__init__.py +34 -0
  253. vllm/lora/layers/base.py +69 -0
  254. vllm/lora/layers/base_linear.py +185 -0
  255. vllm/lora/layers/column_parallel_linear.py +609 -0
  256. vllm/lora/layers/logits_processor.py +247 -0
  257. vllm/lora/layers/qkv_x_parallel_linear.py +8 -0
  258. vllm/lora/layers/replicated_linear.py +60 -0
  259. vllm/lora/layers/row_parallel_linear.py +196 -0
  260. vllm/lora/layers/utils.py +65 -0
  261. vllm/lora/layers/vocal_parallel_embedding.py +174 -0
  262. vllm/lora/lora_weights.py +199 -0
  263. vllm/lora/models.py +816 -0
  264. vllm/lora/ops/__init__.py +0 -0
  265. vllm/lora/ops/ipex_ops/__init__.py +7 -0
  266. vllm/lora/ops/ipex_ops/lora_ops.py +44 -0
  267. vllm/lora/ops/torch_ops/__init__.py +16 -0
  268. vllm/lora/ops/torch_ops/lora_ops.py +119 -0
  269. vllm/lora/ops/triton_ops/__init__.py +12 -0
  270. vllm/lora/ops/triton_ops/kernel_utils.py +243 -0
  271. vllm/lora/ops/triton_ops/lora_expand_op.py +289 -0
  272. vllm/lora/ops/triton_ops/lora_kernel_metadata.py +148 -0
  273. vllm/lora/ops/triton_ops/lora_shrink_op.py +243 -0
  274. vllm/lora/ops/triton_ops/utils.py +126 -0
  275. vllm/lora/ops/xla_ops/__init__.py +7 -0
  276. vllm/lora/ops/xla_ops/lora_ops.py +144 -0
  277. vllm/lora/peft_helper.py +127 -0
  278. vllm/lora/punica_wrapper/__init__.py +10 -0
  279. vllm/lora/punica_wrapper/punica_base.py +458 -0
  280. vllm/lora/punica_wrapper/punica_cpu.py +349 -0
  281. vllm/lora/punica_wrapper/punica_gpu.py +272 -0
  282. vllm/lora/punica_wrapper/punica_selector.py +20 -0
  283. vllm/lora/punica_wrapper/punica_tpu.py +391 -0
  284. vllm/lora/punica_wrapper/punica_xpu.py +276 -0
  285. vllm/lora/punica_wrapper/utils.py +136 -0
  286. vllm/lora/request.py +97 -0
  287. vllm/lora/resolver.py +85 -0
  288. vllm/lora/utils.py +246 -0
  289. vllm/lora/worker_manager.py +267 -0
  290. vllm/model_executor/__init__.py +12 -0
  291. vllm/model_executor/custom_op.py +194 -0
  292. vllm/model_executor/layers/__init__.py +0 -0
  293. vllm/model_executor/layers/activation.py +575 -0
  294. vllm/model_executor/layers/attention_layer_base.py +23 -0
  295. vllm/model_executor/layers/fla/__init__.py +8 -0
  296. vllm/model_executor/layers/fla/ops/__init__.py +17 -0
  297. vllm/model_executor/layers/fla/ops/chunk.py +225 -0
  298. vllm/model_executor/layers/fla/ops/chunk_delta_h.py +290 -0
  299. vllm/model_executor/layers/fla/ops/chunk_o.py +177 -0
  300. vllm/model_executor/layers/fla/ops/chunk_scaled_dot_kkt.py +140 -0
  301. vllm/model_executor/layers/fla/ops/cumsum.py +226 -0
  302. vllm/model_executor/layers/fla/ops/fused_recurrent.py +366 -0
  303. vllm/model_executor/layers/fla/ops/index.py +39 -0
  304. vllm/model_executor/layers/fla/ops/l2norm.py +143 -0
  305. vllm/model_executor/layers/fla/ops/layernorm_guard.py +337 -0
  306. vllm/model_executor/layers/fla/ops/op.py +39 -0
  307. vllm/model_executor/layers/fla/ops/solve_tril.py +365 -0
  308. vllm/model_executor/layers/fla/ops/utils.py +180 -0
  309. vllm/model_executor/layers/fla/ops/wy_fast.py +114 -0
  310. vllm/model_executor/layers/fused_moe/__init__.py +89 -0
  311. vllm/model_executor/layers/fused_moe/batched_deep_gemm_moe.py +322 -0
  312. vllm/model_executor/layers/fused_moe/batched_triton_or_deep_gemm_moe.py +141 -0
  313. vllm/model_executor/layers/fused_moe/config.py +804 -0
  314. vllm/model_executor/layers/fused_moe/configs/E=1,N=14336,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json +146 -0
  315. vllm/model_executor/layers/fused_moe/configs/E=1,N=14336,device_name=NVIDIA_A100-SXM4-80GB.json +146 -0
  316. vllm/model_executor/layers/fused_moe/configs/E=1,N=1792,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json +218 -0
  317. vllm/model_executor/layers/fused_moe/configs/E=1,N=1792,device_name=NVIDIA_A100-SXM4-80GB.json +218 -0
  318. vllm/model_executor/layers/fused_moe/configs/E=1,N=3072,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json +218 -0
  319. vllm/model_executor/layers/fused_moe/configs/E=1,N=3072,device_name=NVIDIA_H100_80GB_HBM3,dtype=int8_w8a16.json +218 -0
  320. vllm/model_executor/layers/fused_moe/configs/E=1,N=3072,device_name=NVIDIA_H100_80GB_HBM3.json +218 -0
  321. vllm/model_executor/layers/fused_moe/configs/E=1,N=3584,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json +218 -0
  322. vllm/model_executor/layers/fused_moe/configs/E=1,N=3584,device_name=NVIDIA_A100-SXM4-80GB.json +218 -0
  323. vllm/model_executor/layers/fused_moe/configs/E=1,N=7168,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json +218 -0
  324. vllm/model_executor/layers/fused_moe/configs/E=1,N=7168,device_name=NVIDIA_A100-SXM4-80GB.json +218 -0
  325. vllm/model_executor/layers/fused_moe/configs/E=128,N=1024,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8.json +164 -0
  326. vllm/model_executor/layers/fused_moe/configs/E=128,N=1024,device_name=AMD_Instinct_MI300X.json +200 -0
  327. vllm/model_executor/layers/fused_moe/configs/E=128,N=1024,device_name=NVIDIA_H100,dtype=fp8_w8a8.json +123 -0
  328. vllm/model_executor/layers/fused_moe/configs/E=128,N=192,device_name=NVIDIA_A100-SXM4-80GB.json +146 -0
  329. vllm/model_executor/layers/fused_moe/configs/E=128,N=192,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  330. vllm/model_executor/layers/fused_moe/configs/E=128,N=192,device_name=NVIDIA_H20-3e.json +146 -0
  331. vllm/model_executor/layers/fused_moe/configs/E=128,N=192,device_name=NVIDIA_H20.json +146 -0
  332. vllm/model_executor/layers/fused_moe/configs/E=128,N=192,device_name=NVIDIA_H200.json +146 -0
  333. vllm/model_executor/layers/fused_moe/configs/E=128,N=352,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json +122 -0
  334. vllm/model_executor/layers/fused_moe/configs/E=128,N=384,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  335. vllm/model_executor/layers/fused_moe/configs/E=128,N=384,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  336. vllm/model_executor/layers/fused_moe/configs/E=128,N=384,device_name=NVIDIA_GB200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  337. vllm/model_executor/layers/fused_moe/configs/E=128,N=384,device_name=NVIDIA_H20,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  338. vllm/model_executor/layers/fused_moe/configs/E=128,N=384,device_name=NVIDIA_H20-3e,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  339. vllm/model_executor/layers/fused_moe/configs/E=128,N=384,device_name=NVIDIA_H20-3e.json +146 -0
  340. vllm/model_executor/layers/fused_moe/configs/E=128,N=384,device_name=NVIDIA_H20.json +146 -0
  341. vllm/model_executor/layers/fused_moe/configs/E=128,N=384,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  342. vllm/model_executor/layers/fused_moe/configs/E=128,N=384,device_name=NVIDIA_H200.json +146 -0
  343. vllm/model_executor/layers/fused_moe/configs/E=128,N=512,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  344. vllm/model_executor/layers/fused_moe/configs/E=128,N=704,device_name=NVIDIA_B200,dtype=fp8_w8a8.json +146 -0
  345. vllm/model_executor/layers/fused_moe/configs/E=128,N=704,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json +114 -0
  346. vllm/model_executor/layers/fused_moe/configs/E=128,N=768,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  347. vllm/model_executor/layers/fused_moe/configs/E=128,N=768,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  348. vllm/model_executor/layers/fused_moe/configs/E=128,N=768,device_name=NVIDIA_GB200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  349. vllm/model_executor/layers/fused_moe/configs/E=128,N=768,device_name=NVIDIA_H20,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  350. vllm/model_executor/layers/fused_moe/configs/E=128,N=768,device_name=NVIDIA_H20-3e,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  351. vllm/model_executor/layers/fused_moe/configs/E=128,N=768,device_name=NVIDIA_H20.json +146 -0
  352. vllm/model_executor/layers/fused_moe/configs/E=128,N=768,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  353. vllm/model_executor/layers/fused_moe/configs/E=128,N=768,device_name=NVIDIA_H200.json +146 -0
  354. vllm/model_executor/layers/fused_moe/configs/E=128,N=96,device_name=NVIDIA_H20.json +146 -0
  355. vllm/model_executor/layers/fused_moe/configs/E=16,N=1024,device_name=AMD_Instinct_MI300X.json +200 -0
  356. vllm/model_executor/layers/fused_moe/configs/E=16,N=1024,device_name=NVIDIA_B200,dtype=fp8_w8a8.json +147 -0
  357. vllm/model_executor/layers/fused_moe/configs/E=16,N=1024,device_name=NVIDIA_B200.json +146 -0
  358. vllm/model_executor/layers/fused_moe/configs/E=16,N=1024,device_name=NVIDIA_H100.json +146 -0
  359. vllm/model_executor/layers/fused_moe/configs/E=16,N=1344,device_name=NVIDIA_A100-SXM4-40GB.json +146 -0
  360. vllm/model_executor/layers/fused_moe/configs/E=16,N=1344,device_name=NVIDIA_A100-SXM4-80GB.json +146 -0
  361. vllm/model_executor/layers/fused_moe/configs/E=16,N=1344,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  362. vllm/model_executor/layers/fused_moe/configs/E=16,N=14336,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json +146 -0
  363. vllm/model_executor/layers/fused_moe/configs/E=16,N=14336,device_name=NVIDIA_A100-SXM4-80GB.json +146 -0
  364. vllm/model_executor/layers/fused_moe/configs/E=16,N=1792,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json +218 -0
  365. vllm/model_executor/layers/fused_moe/configs/E=16,N=1792,device_name=NVIDIA_A100-SXM4-80GB.json +218 -0
  366. vllm/model_executor/layers/fused_moe/configs/E=16,N=2688,device_name=NVIDIA_A100-SXM4-80GB.json +146 -0
  367. vllm/model_executor/layers/fused_moe/configs/E=16,N=2688,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  368. vllm/model_executor/layers/fused_moe/configs/E=16,N=3072,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json +146 -0
  369. vllm/model_executor/layers/fused_moe/configs/E=16,N=3072,device_name=NVIDIA_H100_80GB_HBM3,dtype=int8_w8a16.json +146 -0
  370. vllm/model_executor/layers/fused_moe/configs/E=16,N=3200,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json +130 -0
  371. vllm/model_executor/layers/fused_moe/configs/E=16,N=3584,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json +146 -0
  372. vllm/model_executor/layers/fused_moe/configs/E=16,N=3584,device_name=NVIDIA_A100-SXM4-80GB.json +218 -0
  373. vllm/model_executor/layers/fused_moe/configs/E=16,N=6400,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json +130 -0
  374. vllm/model_executor/layers/fused_moe/configs/E=16,N=7168,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a16.json +146 -0
  375. vllm/model_executor/layers/fused_moe/configs/E=16,N=7168,device_name=NVIDIA_A100-SXM4-80GB.json +146 -0
  376. vllm/model_executor/layers/fused_moe/configs/E=16,N=7168,device_name=NVIDIA_H100_80GB_HBM3,dtype=int8_w8a16.json +146 -0
  377. vllm/model_executor/layers/fused_moe/configs/E=16,N=800,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json +130 -0
  378. vllm/model_executor/layers/fused_moe/configs/E=160,N=192,device_name=NVIDIA_A800-SXM4-80GB.json +146 -0
  379. vllm/model_executor/layers/fused_moe/configs/E=160,N=192,device_name=NVIDIA_H20-3e.json +146 -0
  380. vllm/model_executor/layers/fused_moe/configs/E=160,N=320,device_name=NVIDIA_H20-3e.json +146 -0
  381. vllm/model_executor/layers/fused_moe/configs/E=160,N=384,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8.json +164 -0
  382. vllm/model_executor/layers/fused_moe/configs/E=160,N=640,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  383. vllm/model_executor/layers/fused_moe/configs/E=160,N=640,device_name=NVIDIA_GB200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  384. vllm/model_executor/layers/fused_moe/configs/E=160,N=640,device_name=NVIDIA_H100,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  385. vllm/model_executor/layers/fused_moe/configs/E=20,N=2560,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  386. vllm/model_executor/layers/fused_moe/configs/E=20,N=2560,device_name=NVIDIA_GB200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  387. vllm/model_executor/layers/fused_moe/configs/E=20,N=2560,device_name=NVIDIA_H100,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  388. vllm/model_executor/layers/fused_moe/configs/E=20,N=2560,device_name=NVIDIA_H20-3e,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  389. vllm/model_executor/layers/fused_moe/configs/E=256,N=1024,device_name=AMD_Instinct_MI325X,block_shape=[128,128].json +200 -0
  390. vllm/model_executor/layers/fused_moe/configs/E=256,N=1024,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +200 -0
  391. vllm/model_executor/layers/fused_moe/configs/E=256,N=128,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  392. vllm/model_executor/layers/fused_moe/configs/E=256,N=128,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a8.json +146 -0
  393. vllm/model_executor/layers/fused_moe/configs/E=256,N=128,device_name=NVIDIA_A800-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  394. vllm/model_executor/layers/fused_moe/configs/E=256,N=128,device_name=NVIDIA_A800-SXM4-80GB,dtype=int8_w8a8.json +146 -0
  395. vllm/model_executor/layers/fused_moe/configs/E=256,N=128,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  396. vllm/model_executor/layers/fused_moe/configs/E=256,N=128,device_name=NVIDIA_H20,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  397. vllm/model_executor/layers/fused_moe/configs/E=256,N=128,device_name=NVIDIA_L20Y,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  398. vllm/model_executor/layers/fused_moe/configs/E=256,N=256,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  399. vllm/model_executor/layers/fused_moe/configs/E=256,N=256,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +200 -0
  400. vllm/model_executor/layers/fused_moe/configs/E=256,N=256,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +200 -0
  401. vllm/model_executor/layers/fused_moe/configs/E=256,N=256,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  402. vllm/model_executor/layers/fused_moe/configs/E=256,N=256,device_name=NVIDIA_H20,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  403. vllm/model_executor/layers/fused_moe/configs/E=256,N=256,device_name=NVIDIA_H20,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  404. vllm/model_executor/layers/fused_moe/configs/E=256,N=256,device_name=NVIDIA_H20-3e,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  405. vllm/model_executor/layers/fused_moe/configs/E=256,N=256,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  406. vllm/model_executor/layers/fused_moe/configs/E=256,N=256,device_name=NVIDIA_L20,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  407. vllm/model_executor/layers/fused_moe/configs/E=256,N=512,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +200 -0
  408. vllm/model_executor/layers/fused_moe/configs/E=256,N=512,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  409. vllm/model_executor/layers/fused_moe/configs/E=256,N=64,device_name=NVIDIA_A800-SXM4-80GB.json +146 -0
  410. vllm/model_executor/layers/fused_moe/configs/E=384,N=128,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  411. vllm/model_executor/layers/fused_moe/configs/E=384,N=128,device_name=NVIDIA_GB200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  412. vllm/model_executor/layers/fused_moe/configs/E=384,N=128,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  413. vllm/model_executor/layers/fused_moe/configs/E=384,N=256,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  414. vllm/model_executor/layers/fused_moe/configs/E=384,N=256,device_name=NVIDIA_GB200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  415. vllm/model_executor/layers/fused_moe/configs/E=40,N=2560,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  416. vllm/model_executor/layers/fused_moe/configs/E=40,N=2560,device_name=NVIDIA_GB200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  417. vllm/model_executor/layers/fused_moe/configs/E=40,N=2560,device_name=NVIDIA_H100,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  418. vllm/model_executor/layers/fused_moe/configs/E=512,N=128,device_name=NVIDIA_B200.json +146 -0
  419. vllm/model_executor/layers/fused_moe/configs/E=512,N=128,device_name=NVIDIA_GB200,dtype=fp8_w8a8.json +147 -0
  420. vllm/model_executor/layers/fused_moe/configs/E=512,N=128,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  421. vllm/model_executor/layers/fused_moe/configs/E=512,N=128,device_name=NVIDIA_H20-3e.json +146 -0
  422. vllm/model_executor/layers/fused_moe/configs/E=512,N=128,device_name=NVIDIA_H200.json +146 -0
  423. vllm/model_executor/layers/fused_moe/configs/E=512,N=256,device_name=NVIDIA_B200.json +146 -0
  424. vllm/model_executor/layers/fused_moe/configs/E=512,N=256,device_name=NVIDIA_GB200,dtype=fp8_w8a8.json +146 -0
  425. vllm/model_executor/layers/fused_moe/configs/E=512,N=256,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  426. vllm/model_executor/layers/fused_moe/configs/E=512,N=256,device_name=NVIDIA_H20-3e.json +146 -0
  427. vllm/model_executor/layers/fused_moe/configs/E=512,N=256,device_name=NVIDIA_H200.json +146 -0
  428. vllm/model_executor/layers/fused_moe/configs/E=512,N=512,device_name=NVIDIA_B200.json +146 -0
  429. vllm/model_executor/layers/fused_moe/configs/E=512,N=512,device_name=NVIDIA_GB200,dtype=fp8_w8a8.json +146 -0
  430. vllm/model_executor/layers/fused_moe/configs/E=512,N=512,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  431. vllm/model_executor/layers/fused_moe/configs/E=512,N=512,device_name=NVIDIA_H20-3e.json +146 -0
  432. vllm/model_executor/layers/fused_moe/configs/E=512,N=512,device_name=NVIDIA_H200.json +146 -0
  433. vllm/model_executor/layers/fused_moe/configs/E=512,N=64,device_name=NVIDIA_B200.json +146 -0
  434. vllm/model_executor/layers/fused_moe/configs/E=512,N=64,device_name=NVIDIA_H20-3e.json +146 -0
  435. vllm/model_executor/layers/fused_moe/configs/E=512,N=64,device_name=NVIDIA_H200.json +146 -0
  436. vllm/model_executor/layers/fused_moe/configs/E=60,N=1408,device_name=AMD_Instinct_MI300X.json +200 -0
  437. vllm/model_executor/layers/fused_moe/configs/E=60,N=176,device_name=AMD_Instinct_MI300X.json +200 -0
  438. vllm/model_executor/layers/fused_moe/configs/E=60,N=352,device_name=AMD_Instinct_MI300X.json +200 -0
  439. vllm/model_executor/layers/fused_moe/configs/E=60,N=704,device_name=AMD_Instinct_MI300X.json +200 -0
  440. vllm/model_executor/layers/fused_moe/configs/E=62,N=128,device_name=AMD_Instinct_MI300X.json +200 -0
  441. vllm/model_executor/layers/fused_moe/configs/E=62,N=256,device_name=AMD_Instinct_MI300X.json +200 -0
  442. vllm/model_executor/layers/fused_moe/configs/E=62,N=256,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  443. vllm/model_executor/layers/fused_moe/configs/E=62,N=512,device_name=AMD_Instinct_MI300X.json +200 -0
  444. vllm/model_executor/layers/fused_moe/configs/E=62,N=512,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  445. vllm/model_executor/layers/fused_moe/configs/E=64,N=1280,device_name=NVIDIA_A100-SXM4-80GB.json +146 -0
  446. vllm/model_executor/layers/fused_moe/configs/E=64,N=1280,device_name=NVIDIA_A800-SXM4-80GB.json +146 -0
  447. vllm/model_executor/layers/fused_moe/configs/E=64,N=1280,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json +146 -0
  448. vllm/model_executor/layers/fused_moe/configs/E=64,N=1280,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  449. vllm/model_executor/layers/fused_moe/configs/E=64,N=1280,device_name=NVIDIA_H200,dtype=fp8_w8a8.json +146 -0
  450. vllm/model_executor/layers/fused_moe/configs/E=64,N=1280,device_name=NVIDIA_H200.json +146 -0
  451. vllm/model_executor/layers/fused_moe/configs/E=64,N=1536,device_name=NVIDIA_H20,dtype=fp8_w8a8.json +146 -0
  452. vllm/model_executor/layers/fused_moe/configs/E=64,N=2560,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json +146 -0
  453. vllm/model_executor/layers/fused_moe/configs/E=64,N=2560,device_name=NVIDIA_H200,dtype=fp8_w8a8.json +146 -0
  454. vllm/model_executor/layers/fused_moe/configs/E=64,N=2560,device_name=NVIDIA_H200.json +146 -0
  455. vllm/model_executor/layers/fused_moe/configs/E=64,N=3072,device_name=NVIDIA_H20,dtype=fp8_w8a8.json +146 -0
  456. vllm/model_executor/layers/fused_moe/configs/E=64,N=3072,device_name=NVIDIA_H20.json +146 -0
  457. vllm/model_executor/layers/fused_moe/configs/E=64,N=320,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json +146 -0
  458. vllm/model_executor/layers/fused_moe/configs/E=64,N=320,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  459. vllm/model_executor/layers/fused_moe/configs/E=64,N=320,device_name=NVIDIA_H200,dtype=fp8_w8a8.json +146 -0
  460. vllm/model_executor/layers/fused_moe/configs/E=64,N=320,device_name=NVIDIA_H200.json +146 -0
  461. vllm/model_executor/layers/fused_moe/configs/E=64,N=384,device_name=NVIDIA_H20,dtype=fp8_w8a8.json +146 -0
  462. vllm/model_executor/layers/fused_moe/configs/E=64,N=384,device_name=NVIDIA_H20.json +146 -0
  463. vllm/model_executor/layers/fused_moe/configs/E=64,N=640,device_name=NVIDIA_A100-SXM4-80GB.json +146 -0
  464. vllm/model_executor/layers/fused_moe/configs/E=64,N=640,device_name=NVIDIA_A800-SXM4-80GB.json +146 -0
  465. vllm/model_executor/layers/fused_moe/configs/E=64,N=640,device_name=NVIDIA_GeForce_RTX_4090,dtype=fp8_w8a8.json +146 -0
  466. vllm/model_executor/layers/fused_moe/configs/E=64,N=640,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json +146 -0
  467. vllm/model_executor/layers/fused_moe/configs/E=64,N=640,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  468. vllm/model_executor/layers/fused_moe/configs/E=64,N=640,device_name=NVIDIA_H200,dtype=fp8_w8a8.json +146 -0
  469. vllm/model_executor/layers/fused_moe/configs/E=64,N=640,device_name=NVIDIA_H200.json +146 -0
  470. vllm/model_executor/layers/fused_moe/configs/E=64,N=768,device_name=NVIDIA_H20,dtype=fp8_w8a8.json +146 -0
  471. vllm/model_executor/layers/fused_moe/configs/E=64,N=768,device_name=NVIDIA_H20.json +146 -0
  472. vllm/model_executor/layers/fused_moe/configs/E=64,N=896,device_name=NVIDIA_H20.json +146 -0
  473. vllm/model_executor/layers/fused_moe/configs/E=72,N=192,device_name=AMD_Instinct_MI300X.json +200 -0
  474. vllm/model_executor/layers/fused_moe/configs/E=72,N=384,device_name=AMD_Instinct_MI300X.json +200 -0
  475. vllm/model_executor/layers/fused_moe/configs/E=72,N=384,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  476. vllm/model_executor/layers/fused_moe/configs/E=72,N=768,device_name=AMD_Instinct_MI300X.json +200 -0
  477. vllm/model_executor/layers/fused_moe/configs/E=72,N=768,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  478. vllm/model_executor/layers/fused_moe/configs/E=8,N=14336,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8.json +164 -0
  479. vllm/model_executor/layers/fused_moe/configs/E=8,N=14336,device_name=AMD_Instinct_MI300X.json +200 -0
  480. vllm/model_executor/layers/fused_moe/configs/E=8,N=14336,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8.json +164 -0
  481. vllm/model_executor/layers/fused_moe/configs/E=8,N=14336,device_name=AMD_Instinct_MI325X.json +200 -0
  482. vllm/model_executor/layers/fused_moe/configs/E=8,N=14336,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json +138 -0
  483. vllm/model_executor/layers/fused_moe/configs/E=8,N=14336,device_name=NVIDIA_H200,dtype=fp8_w8a8.json +146 -0
  484. vllm/model_executor/layers/fused_moe/configs/E=8,N=14336,device_name=NVIDIA_H200.json +146 -0
  485. vllm/model_executor/layers/fused_moe/configs/E=8,N=16384,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8.json +164 -0
  486. vllm/model_executor/layers/fused_moe/configs/E=8,N=16384,device_name=AMD_Instinct_MI300X.json +200 -0
  487. vllm/model_executor/layers/fused_moe/configs/E=8,N=16384,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8.json +164 -0
  488. vllm/model_executor/layers/fused_moe/configs/E=8,N=16384,device_name=AMD_Instinct_MI325X.json +200 -0
  489. vllm/model_executor/layers/fused_moe/configs/E=8,N=1792,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8.json +164 -0
  490. vllm/model_executor/layers/fused_moe/configs/E=8,N=1792,device_name=AMD_Instinct_MI300X.json +200 -0
  491. vllm/model_executor/layers/fused_moe/configs/E=8,N=1792,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8.json +164 -0
  492. vllm/model_executor/layers/fused_moe/configs/E=8,N=1792,device_name=AMD_Instinct_MI325X.json +200 -0
  493. vllm/model_executor/layers/fused_moe/configs/E=8,N=1792,device_name=NVIDIA_A100-SXM4-40GB.json +146 -0
  494. vllm/model_executor/layers/fused_moe/configs/E=8,N=1792,device_name=NVIDIA_A100-SXM4-80GB.json +146 -0
  495. vllm/model_executor/layers/fused_moe/configs/E=8,N=1792,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  496. vllm/model_executor/layers/fused_moe/configs/E=8,N=1792,device_name=NVIDIA_H200,dtype=fp8_w8a8.json +146 -0
  497. vllm/model_executor/layers/fused_moe/configs/E=8,N=1792,device_name=NVIDIA_H200.json +146 -0
  498. vllm/model_executor/layers/fused_moe/configs/E=8,N=2048,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8.json +164 -0
  499. vllm/model_executor/layers/fused_moe/configs/E=8,N=2048,device_name=AMD_Instinct_MI300X.json +200 -0
  500. vllm/model_executor/layers/fused_moe/configs/E=8,N=2048,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8.json +164 -0
  501. vllm/model_executor/layers/fused_moe/configs/E=8,N=2048,device_name=AMD_Instinct_MI325X.json +200 -0
  502. vllm/model_executor/layers/fused_moe/configs/E=8,N=2048,device_name=NVIDIA_A100-SXM4-80GB.json +146 -0
  503. vllm/model_executor/layers/fused_moe/configs/E=8,N=2048,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json +146 -0
  504. vllm/model_executor/layers/fused_moe/configs/E=8,N=2048,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  505. vllm/model_executor/layers/fused_moe/configs/E=8,N=2048,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +154 -0
  506. vllm/model_executor/layers/fused_moe/configs/E=8,N=2048,device_name=NVIDIA_H200,dtype=fp8_w8a8.json +146 -0
  507. vllm/model_executor/layers/fused_moe/configs/E=8,N=2048,device_name=NVIDIA_H200.json +146 -0
  508. vllm/model_executor/layers/fused_moe/configs/E=8,N=3584,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8.json +164 -0
  509. vllm/model_executor/layers/fused_moe/configs/E=8,N=3584,device_name=AMD_Instinct_MI300X.json +200 -0
  510. vllm/model_executor/layers/fused_moe/configs/E=8,N=3584,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8.json +164 -0
  511. vllm/model_executor/layers/fused_moe/configs/E=8,N=3584,device_name=AMD_Instinct_MI325X.json +200 -0
  512. vllm/model_executor/layers/fused_moe/configs/E=8,N=3584,device_name=NVIDIA_A100-SXM4-40GB.json +146 -0
  513. vllm/model_executor/layers/fused_moe/configs/E=8,N=3584,device_name=NVIDIA_A100-SXM4-80GB.json +146 -0
  514. vllm/model_executor/layers/fused_moe/configs/E=8,N=3584,device_name=NVIDIA_GeForce_RTX_4090,dtype=fp8_w8a8.json +146 -0
  515. vllm/model_executor/layers/fused_moe/configs/E=8,N=3584,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json +146 -0
  516. vllm/model_executor/layers/fused_moe/configs/E=8,N=3584,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  517. vllm/model_executor/layers/fused_moe/configs/E=8,N=3584,device_name=NVIDIA_H200,dtype=fp8_w8a8.json +146 -0
  518. vllm/model_executor/layers/fused_moe/configs/E=8,N=3584,device_name=NVIDIA_H200.json +146 -0
  519. vllm/model_executor/layers/fused_moe/configs/E=8,N=3584,device_name=NVIDIA_L40S.json +173 -0
  520. vllm/model_executor/layers/fused_moe/configs/E=8,N=4096,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8.json +164 -0
  521. vllm/model_executor/layers/fused_moe/configs/E=8,N=4096,device_name=AMD_Instinct_MI300X.json +200 -0
  522. vllm/model_executor/layers/fused_moe/configs/E=8,N=4096,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8.json +164 -0
  523. vllm/model_executor/layers/fused_moe/configs/E=8,N=4096,device_name=AMD_Instinct_MI325X.json +200 -0
  524. vllm/model_executor/layers/fused_moe/configs/E=8,N=4096,device_name=NVIDIA_A100-SXM4-80GB.json +146 -0
  525. vllm/model_executor/layers/fused_moe/configs/E=8,N=4096,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json +146 -0
  526. vllm/model_executor/layers/fused_moe/configs/E=8,N=4096,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  527. vllm/model_executor/layers/fused_moe/configs/E=8,N=4096,device_name=NVIDIA_H200,dtype=fp8_w8a8.json +146 -0
  528. vllm/model_executor/layers/fused_moe/configs/E=8,N=4096,device_name=NVIDIA_H200.json +146 -0
  529. vllm/model_executor/layers/fused_moe/configs/E=8,N=7168,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8.json +164 -0
  530. vllm/model_executor/layers/fused_moe/configs/E=8,N=7168,device_name=AMD_Instinct_MI300X.json +200 -0
  531. vllm/model_executor/layers/fused_moe/configs/E=8,N=7168,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8.json +164 -0
  532. vllm/model_executor/layers/fused_moe/configs/E=8,N=7168,device_name=AMD_Instinct_MI325X.json +200 -0
  533. vllm/model_executor/layers/fused_moe/configs/E=8,N=7168,device_name=NVIDIA_A100-SXM4-80GB.json +146 -0
  534. vllm/model_executor/layers/fused_moe/configs/E=8,N=7168,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json +146 -0
  535. vllm/model_executor/layers/fused_moe/configs/E=8,N=7168,device_name=NVIDIA_H100_80GB_HBM3.json +146 -0
  536. vllm/model_executor/layers/fused_moe/configs/E=8,N=7168,device_name=NVIDIA_H200,dtype=fp8_w8a8.json +146 -0
  537. vllm/model_executor/layers/fused_moe/configs/E=8,N=7168,device_name=NVIDIA_H200.json +146 -0
  538. vllm/model_executor/layers/fused_moe/configs/E=8,N=8192,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8.json +164 -0
  539. vllm/model_executor/layers/fused_moe/configs/E=8,N=8192,device_name=AMD_Instinct_MI300X.json +200 -0
  540. vllm/model_executor/layers/fused_moe/configs/E=8,N=8192,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8.json +164 -0
  541. vllm/model_executor/layers/fused_moe/configs/E=8,N=8192,device_name=AMD_Instinct_MI325X.json +200 -0
  542. vllm/model_executor/layers/fused_moe/configs/E=8,N=8192,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8.json +146 -0
  543. vllm/model_executor/layers/fused_moe/configs/E=8,N=8192,device_name=NVIDIA_H200,dtype=fp8_w8a8.json +146 -0
  544. vllm/model_executor/layers/fused_moe/configs/README +12 -0
  545. vllm/model_executor/layers/fused_moe/cpu_fused_moe.py +300 -0
  546. vllm/model_executor/layers/fused_moe/cutlass_moe.py +957 -0
  547. vllm/model_executor/layers/fused_moe/deep_gemm_moe.py +362 -0
  548. vllm/model_executor/layers/fused_moe/deep_gemm_utils.py +413 -0
  549. vllm/model_executor/layers/fused_moe/deepep_ht_prepare_finalize.py +361 -0
  550. vllm/model_executor/layers/fused_moe/deepep_ll_prepare_finalize.py +274 -0
  551. vllm/model_executor/layers/fused_moe/flashinfer_cutlass_moe.py +268 -0
  552. vllm/model_executor/layers/fused_moe/flashinfer_cutlass_prepare_finalize.py +300 -0
  553. vllm/model_executor/layers/fused_moe/flashinfer_trtllm_moe.py +184 -0
  554. vllm/model_executor/layers/fused_moe/fused_batched_moe.py +993 -0
  555. vllm/model_executor/layers/fused_moe/fused_marlin_moe.py +239 -0
  556. vllm/model_executor/layers/fused_moe/fused_moe.py +1890 -0
  557. vllm/model_executor/layers/fused_moe/gpt_oss_triton_kernels_moe.py +307 -0
  558. vllm/model_executor/layers/fused_moe/layer.py +2195 -0
  559. vllm/model_executor/layers/fused_moe/modular_kernel.py +1038 -0
  560. vllm/model_executor/layers/fused_moe/moe_align_block_size.py +87 -0
  561. vllm/model_executor/layers/fused_moe/moe_pallas.py +80 -0
  562. vllm/model_executor/layers/fused_moe/moe_permute_unpermute.py +205 -0
  563. vllm/model_executor/layers/fused_moe/moe_torch_iterative.py +60 -0
  564. vllm/model_executor/layers/fused_moe/pplx_prepare_finalize.py +341 -0
  565. vllm/model_executor/layers/fused_moe/prepare_finalize.py +70 -0
  566. vllm/model_executor/layers/fused_moe/rocm_aiter_fused_moe.py +424 -0
  567. vllm/model_executor/layers/fused_moe/routing_simulator.py +291 -0
  568. vllm/model_executor/layers/fused_moe/topk_weight_and_reduce.py +146 -0
  569. vllm/model_executor/layers/fused_moe/triton_deep_gemm_moe.py +143 -0
  570. vllm/model_executor/layers/fused_moe/trtllm_moe.py +191 -0
  571. vllm/model_executor/layers/fused_moe/utils.py +274 -0
  572. vllm/model_executor/layers/layernorm.py +395 -0
  573. vllm/model_executor/layers/lightning_attn.py +661 -0
  574. vllm/model_executor/layers/linear.py +1603 -0
  575. vllm/model_executor/layers/logits_processor.py +106 -0
  576. vllm/model_executor/layers/mamba/__init__.py +0 -0
  577. vllm/model_executor/layers/mamba/abstract.py +42 -0
  578. vllm/model_executor/layers/mamba/linear_attn.py +403 -0
  579. vllm/model_executor/layers/mamba/mamba_mixer.py +466 -0
  580. vllm/model_executor/layers/mamba/mamba_mixer2.py +764 -0
  581. vllm/model_executor/layers/mamba/mamba_utils.py +186 -0
  582. vllm/model_executor/layers/mamba/ops/__init__.py +0 -0
  583. vllm/model_executor/layers/mamba/ops/causal_conv1d.py +1092 -0
  584. vllm/model_executor/layers/mamba/ops/layernorm_gated.py +168 -0
  585. vllm/model_executor/layers/mamba/ops/mamba_ssm.py +414 -0
  586. vllm/model_executor/layers/mamba/ops/ssd_bmm.py +242 -0
  587. vllm/model_executor/layers/mamba/ops/ssd_chunk_scan.py +527 -0
  588. vllm/model_executor/layers/mamba/ops/ssd_chunk_state.py +724 -0
  589. vllm/model_executor/layers/mamba/ops/ssd_combined.py +238 -0
  590. vllm/model_executor/layers/mamba/ops/ssd_state_passing.py +200 -0
  591. vllm/model_executor/layers/mamba/short_conv.py +253 -0
  592. vllm/model_executor/layers/mla.py +173 -0
  593. vllm/model_executor/layers/pooler.py +719 -0
  594. vllm/model_executor/layers/quantization/__init__.py +157 -0
  595. vllm/model_executor/layers/quantization/auto_round.py +388 -0
  596. vllm/model_executor/layers/quantization/awq.py +228 -0
  597. vllm/model_executor/layers/quantization/awq_marlin.py +554 -0
  598. vllm/model_executor/layers/quantization/awq_triton.py +320 -0
  599. vllm/model_executor/layers/quantization/base_config.py +170 -0
  600. vllm/model_executor/layers/quantization/bitblas.py +464 -0
  601. vllm/model_executor/layers/quantization/bitsandbytes.py +627 -0
  602. vllm/model_executor/layers/quantization/compressed_tensors/__init__.py +0 -0
  603. vllm/model_executor/layers/quantization/compressed_tensors/compressed_tensors.py +797 -0
  604. vllm/model_executor/layers/quantization/compressed_tensors/compressed_tensors_moe.py +2074 -0
  605. vllm/model_executor/layers/quantization/compressed_tensors/schemes/__init__.py +27 -0
  606. vllm/model_executor/layers/quantization/compressed_tensors/schemes/compressed_tensors_24.py +366 -0
  607. vllm/model_executor/layers/quantization/compressed_tensors/schemes/compressed_tensors_scheme.py +55 -0
  608. vllm/model_executor/layers/quantization/compressed_tensors/schemes/compressed_tensors_w4a16_24.py +160 -0
  609. vllm/model_executor/layers/quantization/compressed_tensors/schemes/compressed_tensors_w4a16_nvfp4.py +105 -0
  610. vllm/model_executor/layers/quantization/compressed_tensors/schemes/compressed_tensors_w4a4_nvfp4.py +185 -0
  611. vllm/model_executor/layers/quantization/compressed_tensors/schemes/compressed_tensors_w4a8_fp8.py +169 -0
  612. vllm/model_executor/layers/quantization/compressed_tensors/schemes/compressed_tensors_w4a8_int.py +135 -0
  613. vllm/model_executor/layers/quantization/compressed_tensors/schemes/compressed_tensors_w8a16_fp8.py +121 -0
  614. vllm/model_executor/layers/quantization/compressed_tensors/schemes/compressed_tensors_w8a8_fp8.py +157 -0
  615. vllm/model_executor/layers/quantization/compressed_tensors/schemes/compressed_tensors_w8a8_int8.py +111 -0
  616. vllm/model_executor/layers/quantization/compressed_tensors/schemes/compressed_tensors_wNa16.py +201 -0
  617. vllm/model_executor/layers/quantization/compressed_tensors/transform/__init__.py +0 -0
  618. vllm/model_executor/layers/quantization/compressed_tensors/transform/linear.py +238 -0
  619. vllm/model_executor/layers/quantization/compressed_tensors/transform/module.py +153 -0
  620. vllm/model_executor/layers/quantization/compressed_tensors/transform/schemes/__init__.py +0 -0
  621. vllm/model_executor/layers/quantization/compressed_tensors/transform/schemes/linear_qutlass_nvfp4.py +46 -0
  622. vllm/model_executor/layers/quantization/compressed_tensors/transform/utils.py +13 -0
  623. vllm/model_executor/layers/quantization/compressed_tensors/triton_scaled_mm.py +206 -0
  624. vllm/model_executor/layers/quantization/compressed_tensors/utils.py +216 -0
  625. vllm/model_executor/layers/quantization/deepspeedfp.py +196 -0
  626. vllm/model_executor/layers/quantization/experts_int8.py +223 -0
  627. vllm/model_executor/layers/quantization/fbgemm_fp8.py +172 -0
  628. vllm/model_executor/layers/quantization/fp8.py +1098 -0
  629. vllm/model_executor/layers/quantization/gguf.py +599 -0
  630. vllm/model_executor/layers/quantization/gptq.py +340 -0
  631. vllm/model_executor/layers/quantization/gptq_bitblas.py +448 -0
  632. vllm/model_executor/layers/quantization/gptq_marlin.py +751 -0
  633. vllm/model_executor/layers/quantization/gptq_marlin_24.py +297 -0
  634. vllm/model_executor/layers/quantization/hqq_marlin.py +333 -0
  635. vllm/model_executor/layers/quantization/inc.py +61 -0
  636. vllm/model_executor/layers/quantization/input_quant_fp8.py +156 -0
  637. vllm/model_executor/layers/quantization/ipex_quant.py +415 -0
  638. vllm/model_executor/layers/quantization/kernels/__init__.py +0 -0
  639. vllm/model_executor/layers/quantization/kernels/mixed_precision/MPLinearKernel.py +91 -0
  640. vllm/model_executor/layers/quantization/kernels/mixed_precision/__init__.py +93 -0
  641. vllm/model_executor/layers/quantization/kernels/mixed_precision/allspark.py +116 -0
  642. vllm/model_executor/layers/quantization/kernels/mixed_precision/bitblas.py +302 -0
  643. vllm/model_executor/layers/quantization/kernels/mixed_precision/conch.py +92 -0
  644. vllm/model_executor/layers/quantization/kernels/mixed_precision/cutlass.py +117 -0
  645. vllm/model_executor/layers/quantization/kernels/mixed_precision/dynamic_4bit.py +92 -0
  646. vllm/model_executor/layers/quantization/kernels/mixed_precision/exllama.py +143 -0
  647. vllm/model_executor/layers/quantization/kernels/mixed_precision/machete.py +144 -0
  648. vllm/model_executor/layers/quantization/kernels/mixed_precision/marlin.py +139 -0
  649. vllm/model_executor/layers/quantization/kernels/scaled_mm/ScaledMMLinearKernel.py +67 -0
  650. vllm/model_executor/layers/quantization/kernels/scaled_mm/__init__.py +89 -0
  651. vllm/model_executor/layers/quantization/kernels/scaled_mm/aiter.py +161 -0
  652. vllm/model_executor/layers/quantization/kernels/scaled_mm/cpu.py +206 -0
  653. vllm/model_executor/layers/quantization/kernels/scaled_mm/cutlass.py +137 -0
  654. vllm/model_executor/layers/quantization/kernels/scaled_mm/triton.py +41 -0
  655. vllm/model_executor/layers/quantization/kernels/scaled_mm/xla.py +104 -0
  656. vllm/model_executor/layers/quantization/kv_cache.py +143 -0
  657. vllm/model_executor/layers/quantization/modelopt.py +1596 -0
  658. vllm/model_executor/layers/quantization/moe_wna16.py +484 -0
  659. vllm/model_executor/layers/quantization/mxfp4.py +988 -0
  660. vllm/model_executor/layers/quantization/petit.py +306 -0
  661. vllm/model_executor/layers/quantization/ptpc_fp8.py +129 -0
  662. vllm/model_executor/layers/quantization/quark/__init__.py +0 -0
  663. vllm/model_executor/layers/quantization/quark/quark.py +432 -0
  664. vllm/model_executor/layers/quantization/quark/quark_moe.py +561 -0
  665. vllm/model_executor/layers/quantization/quark/schemes/__init__.py +9 -0
  666. vllm/model_executor/layers/quantization/quark/schemes/quark_scheme.py +55 -0
  667. vllm/model_executor/layers/quantization/quark/schemes/quark_w4a4_mxfp4.py +239 -0
  668. vllm/model_executor/layers/quantization/quark/schemes/quark_w8a8_fp8.py +163 -0
  669. vllm/model_executor/layers/quantization/quark/schemes/quark_w8a8_int8.py +122 -0
  670. vllm/model_executor/layers/quantization/quark/utils.py +105 -0
  671. vllm/model_executor/layers/quantization/rtn.py +466 -0
  672. vllm/model_executor/layers/quantization/schema.py +86 -0
  673. vllm/model_executor/layers/quantization/torchao.py +214 -0
  674. vllm/model_executor/layers/quantization/tpu_int8.py +125 -0
  675. vllm/model_executor/layers/quantization/utils/__init__.py +6 -0
  676. vllm/model_executor/layers/quantization/utils/allspark_utils.py +52 -0
  677. vllm/model_executor/layers/quantization/utils/bitblas_utils.py +210 -0
  678. vllm/model_executor/layers/quantization/utils/configs/N=12288,K=7168,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  679. vllm/model_executor/layers/quantization/utils/configs/N=12288,K=7168,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  680. vllm/model_executor/layers/quantization/utils/configs/N=1536,K=1536,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  681. vllm/model_executor/layers/quantization/utils/configs/N=1536,K=1536,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  682. vllm/model_executor/layers/quantization/utils/configs/N=1536,K=1536,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  683. vllm/model_executor/layers/quantization/utils/configs/N=1536,K=1536,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  684. vllm/model_executor/layers/quantization/utils/configs/N=1536,K=1536,device_name=NVIDIA_A800-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  685. vllm/model_executor/layers/quantization/utils/configs/N=1536,K=1536,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  686. vllm/model_executor/layers/quantization/utils/configs/N=1536,K=1536,device_name=NVIDIA_H20,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  687. vllm/model_executor/layers/quantization/utils/configs/N=1536,K=1536,device_name=NVIDIA_L20Y,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  688. vllm/model_executor/layers/quantization/utils/configs/N=1536,K=7168,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  689. vllm/model_executor/layers/quantization/utils/configs/N=1536,K=7168,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  690. vllm/model_executor/layers/quantization/utils/configs/N=1536,K=7168,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  691. vllm/model_executor/layers/quantization/utils/configs/N=1536,K=7168,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  692. vllm/model_executor/layers/quantization/utils/configs/N=1536,K=7168,device_name=NVIDIA_A800-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  693. vllm/model_executor/layers/quantization/utils/configs/N=1536,K=7168,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  694. vllm/model_executor/layers/quantization/utils/configs/N=1536,K=7168,device_name=NVIDIA_H20,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  695. vllm/model_executor/layers/quantization/utils/configs/N=1536,K=7168,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  696. vllm/model_executor/layers/quantization/utils/configs/N=1536,K=7168,device_name=NVIDIA_L20,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  697. vllm/model_executor/layers/quantization/utils/configs/N=1536,K=7168,device_name=NVIDIA_L20Y,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  698. vllm/model_executor/layers/quantization/utils/configs/N=2048,K=512,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  699. vllm/model_executor/layers/quantization/utils/configs/N=2048,K=512,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  700. vllm/model_executor/layers/quantization/utils/configs/N=2048,K=512,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  701. vllm/model_executor/layers/quantization/utils/configs/N=2048,K=512,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  702. vllm/model_executor/layers/quantization/utils/configs/N=2048,K=512,device_name=NVIDIA_A800-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  703. vllm/model_executor/layers/quantization/utils/configs/N=2048,K=512,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  704. vllm/model_executor/layers/quantization/utils/configs/N=2048,K=512,device_name=NVIDIA_H20,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  705. vllm/model_executor/layers/quantization/utils/configs/N=2048,K=512,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  706. vllm/model_executor/layers/quantization/utils/configs/N=2048,K=512,device_name=NVIDIA_L20Y,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  707. vllm/model_executor/layers/quantization/utils/configs/N=2112,K=7168,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  708. vllm/model_executor/layers/quantization/utils/configs/N=2112,K=7168,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  709. vllm/model_executor/layers/quantization/utils/configs/N=2304,K=7168,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  710. vllm/model_executor/layers/quantization/utils/configs/N=2304,K=7168,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  711. vllm/model_executor/layers/quantization/utils/configs/N=2304,K=7168,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  712. vllm/model_executor/layers/quantization/utils/configs/N=2304,K=7168,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  713. vllm/model_executor/layers/quantization/utils/configs/N=2304,K=7168,device_name=NVIDIA_A800-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  714. vllm/model_executor/layers/quantization/utils/configs/N=2304,K=7168,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  715. vllm/model_executor/layers/quantization/utils/configs/N=2304,K=7168,device_name=NVIDIA_H20,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  716. vllm/model_executor/layers/quantization/utils/configs/N=2304,K=7168,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  717. vllm/model_executor/layers/quantization/utils/configs/N=2304,K=7168,device_name=NVIDIA_L20Y,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  718. vllm/model_executor/layers/quantization/utils/configs/N=24576,K=1536,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  719. vllm/model_executor/layers/quantization/utils/configs/N=24576,K=1536,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  720. vllm/model_executor/layers/quantization/utils/configs/N=24576,K=7168,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  721. vllm/model_executor/layers/quantization/utils/configs/N=24576,K=7168,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  722. vllm/model_executor/layers/quantization/utils/configs/N=24576,K=7168,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  723. vllm/model_executor/layers/quantization/utils/configs/N=24576,K=7168,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  724. vllm/model_executor/layers/quantization/utils/configs/N=24576,K=7168,device_name=NVIDIA_A800-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  725. vllm/model_executor/layers/quantization/utils/configs/N=24576,K=7168,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  726. vllm/model_executor/layers/quantization/utils/configs/N=24576,K=7168,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  727. vllm/model_executor/layers/quantization/utils/configs/N=24576,K=7168,device_name=NVIDIA_H20,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  728. vllm/model_executor/layers/quantization/utils/configs/N=24576,K=7168,device_name=NVIDIA_H20,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  729. vllm/model_executor/layers/quantization/utils/configs/N=24576,K=7168,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  730. vllm/model_executor/layers/quantization/utils/configs/N=24576,K=7168,device_name=NVIDIA_L20,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  731. vllm/model_executor/layers/quantization/utils/configs/N=24576,K=7168,device_name=NVIDIA_L20Y,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  732. vllm/model_executor/layers/quantization/utils/configs/N=256,K=7168,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  733. vllm/model_executor/layers/quantization/utils/configs/N=256,K=7168,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  734. vllm/model_executor/layers/quantization/utils/configs/N=256,K=7168,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  735. vllm/model_executor/layers/quantization/utils/configs/N=256,K=7168,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  736. vllm/model_executor/layers/quantization/utils/configs/N=256,K=7168,device_name=NVIDIA_A800-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  737. vllm/model_executor/layers/quantization/utils/configs/N=256,K=7168,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  738. vllm/model_executor/layers/quantization/utils/configs/N=256,K=7168,device_name=NVIDIA_H20,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  739. vllm/model_executor/layers/quantization/utils/configs/N=256,K=7168,device_name=NVIDIA_L20Y,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  740. vllm/model_executor/layers/quantization/utils/configs/N=3072,K=1536,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  741. vllm/model_executor/layers/quantization/utils/configs/N=3072,K=1536,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  742. vllm/model_executor/layers/quantization/utils/configs/N=3072,K=1536,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  743. vllm/model_executor/layers/quantization/utils/configs/N=3072,K=1536,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  744. vllm/model_executor/layers/quantization/utils/configs/N=3072,K=1536,device_name=NVIDIA_H20,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  745. vllm/model_executor/layers/quantization/utils/configs/N=3072,K=1536,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  746. vllm/model_executor/layers/quantization/utils/configs/N=3072,K=1536,device_name=NVIDIA_L20,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  747. vllm/model_executor/layers/quantization/utils/configs/N=3072,K=7168,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  748. vllm/model_executor/layers/quantization/utils/configs/N=3072,K=7168,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  749. vllm/model_executor/layers/quantization/utils/configs/N=3072,K=7168,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  750. vllm/model_executor/layers/quantization/utils/configs/N=3072,K=7168,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  751. vllm/model_executor/layers/quantization/utils/configs/N=3072,K=7168,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  752. vllm/model_executor/layers/quantization/utils/configs/N=3072,K=7168,device_name=NVIDIA_H20,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  753. vllm/model_executor/layers/quantization/utils/configs/N=3072,K=7168,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  754. vllm/model_executor/layers/quantization/utils/configs/N=3072,K=7168,device_name=NVIDIA_L20,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  755. vllm/model_executor/layers/quantization/utils/configs/N=32768,K=512,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  756. vllm/model_executor/layers/quantization/utils/configs/N=32768,K=512,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  757. vllm/model_executor/layers/quantization/utils/configs/N=32768,K=512,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  758. vllm/model_executor/layers/quantization/utils/configs/N=32768,K=512,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  759. vllm/model_executor/layers/quantization/utils/configs/N=32768,K=512,device_name=NVIDIA_A800-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  760. vllm/model_executor/layers/quantization/utils/configs/N=32768,K=512,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  761. vllm/model_executor/layers/quantization/utils/configs/N=32768,K=512,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  762. vllm/model_executor/layers/quantization/utils/configs/N=32768,K=512,device_name=NVIDIA_H20,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  763. vllm/model_executor/layers/quantization/utils/configs/N=32768,K=512,device_name=NVIDIA_H20,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  764. vllm/model_executor/layers/quantization/utils/configs/N=32768,K=512,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  765. vllm/model_executor/layers/quantization/utils/configs/N=32768,K=512,device_name=NVIDIA_L20,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  766. vllm/model_executor/layers/quantization/utils/configs/N=32768,K=512,device_name=NVIDIA_L20Y,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  767. vllm/model_executor/layers/quantization/utils/configs/N=36864,K=7168,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  768. vllm/model_executor/layers/quantization/utils/configs/N=36864,K=7168,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  769. vllm/model_executor/layers/quantization/utils/configs/N=36864,K=7168,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  770. vllm/model_executor/layers/quantization/utils/configs/N=36864,K=7168,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  771. vllm/model_executor/layers/quantization/utils/configs/N=36864,K=7168,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  772. vllm/model_executor/layers/quantization/utils/configs/N=4096,K=512,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  773. vllm/model_executor/layers/quantization/utils/configs/N=4096,K=512,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  774. vllm/model_executor/layers/quantization/utils/configs/N=4096,K=512,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  775. vllm/model_executor/layers/quantization/utils/configs/N=4096,K=512,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  776. vllm/model_executor/layers/quantization/utils/configs/N=4096,K=512,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  777. vllm/model_executor/layers/quantization/utils/configs/N=4096,K=512,device_name=NVIDIA_H20,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  778. vllm/model_executor/layers/quantization/utils/configs/N=4096,K=512,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  779. vllm/model_executor/layers/quantization/utils/configs/N=4096,K=512,device_name=NVIDIA_L20,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  780. vllm/model_executor/layers/quantization/utils/configs/N=4096,K=7168,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  781. vllm/model_executor/layers/quantization/utils/configs/N=4096,K=7168,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  782. vllm/model_executor/layers/quantization/utils/configs/N=4608,K=7168,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  783. vllm/model_executor/layers/quantization/utils/configs/N=4608,K=7168,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  784. vllm/model_executor/layers/quantization/utils/configs/N=4608,K=7168,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  785. vllm/model_executor/layers/quantization/utils/configs/N=4608,K=7168,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  786. vllm/model_executor/layers/quantization/utils/configs/N=4608,K=7168,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  787. vllm/model_executor/layers/quantization/utils/configs/N=4608,K=7168,device_name=NVIDIA_H20,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  788. vllm/model_executor/layers/quantization/utils/configs/N=4608,K=7168,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  789. vllm/model_executor/layers/quantization/utils/configs/N=4608,K=7168,device_name=NVIDIA_L20,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  790. vllm/model_executor/layers/quantization/utils/configs/N=512,K=7168,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  791. vllm/model_executor/layers/quantization/utils/configs/N=512,K=7168,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  792. vllm/model_executor/layers/quantization/utils/configs/N=512,K=7168,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  793. vllm/model_executor/layers/quantization/utils/configs/N=512,K=7168,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  794. vllm/model_executor/layers/quantization/utils/configs/N=512,K=7168,device_name=NVIDIA_H20,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  795. vllm/model_executor/layers/quantization/utils/configs/N=512,K=7168,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  796. vllm/model_executor/layers/quantization/utils/configs/N=512,K=7168,device_name=NVIDIA_L20,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  797. vllm/model_executor/layers/quantization/utils/configs/N=576,K=7168,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  798. vllm/model_executor/layers/quantization/utils/configs/N=576,K=7168,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  799. vllm/model_executor/layers/quantization/utils/configs/N=576,K=7168,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  800. vllm/model_executor/layers/quantization/utils/configs/N=576,K=7168,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  801. vllm/model_executor/layers/quantization/utils/configs/N=576,K=7168,device_name=NVIDIA_A800-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  802. vllm/model_executor/layers/quantization/utils/configs/N=576,K=7168,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  803. vllm/model_executor/layers/quantization/utils/configs/N=576,K=7168,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  804. vllm/model_executor/layers/quantization/utils/configs/N=576,K=7168,device_name=NVIDIA_H20,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  805. vllm/model_executor/layers/quantization/utils/configs/N=576,K=7168,device_name=NVIDIA_H20,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  806. vllm/model_executor/layers/quantization/utils/configs/N=576,K=7168,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  807. vllm/model_executor/layers/quantization/utils/configs/N=576,K=7168,device_name=NVIDIA_L20,dtype=fp8_w8a8,block_shape=[128,128].json +18 -0
  808. vllm/model_executor/layers/quantization/utils/configs/N=576,K=7168,device_name=NVIDIA_L20Y,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  809. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=1024,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  810. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=1024,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  811. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=1024,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  812. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=1024,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  813. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=1024,device_name=NVIDIA_A800-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  814. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=1024,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  815. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=1024,device_name=NVIDIA_H20,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  816. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=1024,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  817. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=1024,device_name=NVIDIA_L20Y,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  818. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=1152,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  819. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=1152,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  820. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=1152,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  821. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=1152,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  822. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=1152,device_name=NVIDIA_A800-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  823. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=1152,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  824. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=1152,device_name=NVIDIA_H20,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  825. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=1152,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  826. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=1152,device_name=NVIDIA_L20Y,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  827. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=128,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  828. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=128,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  829. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=128,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  830. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=128,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  831. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=128,device_name=NVIDIA_A800-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  832. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=128,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  833. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=128,device_name=NVIDIA_H20,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  834. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=128,device_name=NVIDIA_L20Y,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  835. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=16384,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  836. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=16384,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  837. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=16384,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  838. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=16384,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  839. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=16384,device_name=NVIDIA_A800-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  840. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=16384,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  841. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=16384,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  842. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=16384,device_name=NVIDIA_H20,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  843. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=16384,device_name=NVIDIA_H20,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  844. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=16384,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  845. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=16384,device_name=NVIDIA_L20,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  846. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=16384,device_name=NVIDIA_L20Y,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  847. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=18432,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  848. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=18432,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  849. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=18432,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  850. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=18432,device_name=NVIDIA_A100-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  851. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=18432,device_name=NVIDIA_A800-SXM4-80GB,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  852. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=18432,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  853. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=18432,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  854. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=18432,device_name=NVIDIA_H20,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  855. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=18432,device_name=NVIDIA_H20,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  856. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=18432,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  857. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=18432,device_name=NVIDIA_L20,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  858. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=18432,device_name=NVIDIA_L20Y,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  859. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=2048,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  860. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=2048,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  861. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=2048,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  862. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=2048,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  863. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=2048,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  864. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=2048,device_name=NVIDIA_H20,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  865. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=2048,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  866. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=2048,device_name=NVIDIA_L20,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  867. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=2304,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  868. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=2304,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  869. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=2304,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  870. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=2304,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  871. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=2304,device_name=NVIDIA_H100_80GB_HBM3,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  872. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=2304,device_name=NVIDIA_H20,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  873. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=2304,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  874. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=2304,device_name=NVIDIA_L20,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  875. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=256,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  876. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=256,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  877. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=256,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  878. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=256,device_name=NVIDIA_B200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  879. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=256,device_name=NVIDIA_H20,dtype=int8_w8a8,block_shape=[128,128].json +146 -0
  880. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=256,device_name=NVIDIA_H200,dtype=fp8_w8a8,block_shape=[128,128].json +146 -0
  881. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=256,device_name=NVIDIA_L20,dtype=fp8_w8a8,block_shape=[128,128].json +26 -0
  882. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=8192,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  883. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=8192,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  884. vllm/model_executor/layers/quantization/utils/configs/N=7168,K=8192,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  885. vllm/model_executor/layers/quantization/utils/configs/N=8192,K=1536,device_name=AMD_Instinct_MI300X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  886. vllm/model_executor/layers/quantization/utils/configs/N=8192,K=1536,device_name=AMD_Instinct_MI325X,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  887. vllm/model_executor/layers/quantization/utils/configs/N=8192,K=1536,device_name=AMD_Instinct_MI325_OAM,dtype=fp8_w8a8,block_shape=[128,128].json +164 -0
  888. vllm/model_executor/layers/quantization/utils/configs/README.md +3 -0
  889. vllm/model_executor/layers/quantization/utils/flashinfer_fp4_moe.py +79 -0
  890. vllm/model_executor/layers/quantization/utils/flashinfer_utils.py +248 -0
  891. vllm/model_executor/layers/quantization/utils/fp8_utils.py +949 -0
  892. vllm/model_executor/layers/quantization/utils/gptq_utils.py +146 -0
  893. vllm/model_executor/layers/quantization/utils/int8_utils.py +492 -0
  894. vllm/model_executor/layers/quantization/utils/layer_utils.py +40 -0
  895. vllm/model_executor/layers/quantization/utils/machete_utils.py +50 -0
  896. vllm/model_executor/layers/quantization/utils/marlin_utils.py +479 -0
  897. vllm/model_executor/layers/quantization/utils/marlin_utils_fp4.py +396 -0
  898. vllm/model_executor/layers/quantization/utils/marlin_utils_fp8.py +345 -0
  899. vllm/model_executor/layers/quantization/utils/marlin_utils_test.py +165 -0
  900. vllm/model_executor/layers/quantization/utils/marlin_utils_test_24.py +464 -0
  901. vllm/model_executor/layers/quantization/utils/mxfp4_utils.py +141 -0
  902. vllm/model_executor/layers/quantization/utils/mxfp8_utils.py +20 -0
  903. vllm/model_executor/layers/quantization/utils/nvfp4_emulation_utils.py +137 -0
  904. vllm/model_executor/layers/quantization/utils/nvfp4_moe_support.py +59 -0
  905. vllm/model_executor/layers/quantization/utils/petit_utils.py +122 -0
  906. vllm/model_executor/layers/quantization/utils/quant_utils.py +641 -0
  907. vllm/model_executor/layers/quantization/utils/w8a8_utils.py +458 -0
  908. vllm/model_executor/layers/resampler.py +270 -0
  909. vllm/model_executor/layers/rotary_embedding/__init__.py +204 -0
  910. vllm/model_executor/layers/rotary_embedding/base.py +177 -0
  911. vllm/model_executor/layers/rotary_embedding/common.py +150 -0
  912. vllm/model_executor/layers/rotary_embedding/deepseek_scaling_rope.py +138 -0
  913. vllm/model_executor/layers/rotary_embedding/dual_chunk_rope.py +197 -0
  914. vllm/model_executor/layers/rotary_embedding/dynamic_ntk_alpha_rope.py +41 -0
  915. vllm/model_executor/layers/rotary_embedding/dynamic_ntk_scaling_rope.py +67 -0
  916. vllm/model_executor/layers/rotary_embedding/ernie45_vl_rope.py +80 -0
  917. vllm/model_executor/layers/rotary_embedding/linear_scaling_rope.py +115 -0
  918. vllm/model_executor/layers/rotary_embedding/llama3_rope.py +54 -0
  919. vllm/model_executor/layers/rotary_embedding/llama4_vision_rope.py +81 -0
  920. vllm/model_executor/layers/rotary_embedding/mrope.py +1321 -0
  921. vllm/model_executor/layers/rotary_embedding/ntk_scaling_rope.py +42 -0
  922. vllm/model_executor/layers/rotary_embedding/phi3_long_rope_scaled_rope.py +129 -0
  923. vllm/model_executor/layers/rotary_embedding/rocm_aiter_rope_ops.py +86 -0
  924. vllm/model_executor/layers/rotary_embedding/yarn_scaling_rope.py +68 -0
  925. vllm/model_executor/layers/shared_fused_moe/__init__.py +6 -0
  926. vllm/model_executor/layers/shared_fused_moe/shared_fused_moe.py +56 -0
  927. vllm/model_executor/layers/utils.py +195 -0
  928. vllm/model_executor/layers/vocab_parallel_embedding.py +487 -0
  929. vllm/model_executor/model_loader/__init__.py +138 -0
  930. vllm/model_executor/model_loader/base_loader.py +52 -0
  931. vllm/model_executor/model_loader/bitsandbytes_loader.py +788 -0
  932. vllm/model_executor/model_loader/default_loader.py +277 -0
  933. vllm/model_executor/model_loader/dummy_loader.py +28 -0
  934. vllm/model_executor/model_loader/gguf_loader.py +155 -0
  935. vllm/model_executor/model_loader/runai_streamer_loader.py +104 -0
  936. vllm/model_executor/model_loader/sharded_state_loader.py +199 -0
  937. vllm/model_executor/model_loader/tensorizer.py +738 -0
  938. vllm/model_executor/model_loader/tensorizer_loader.py +143 -0
  939. vllm/model_executor/model_loader/tpu.py +114 -0
  940. vllm/model_executor/model_loader/utils.py +292 -0
  941. vllm/model_executor/model_loader/weight_utils.py +990 -0
  942. vllm/model_executor/models/__init__.py +33 -0
  943. vllm/model_executor/models/adapters.py +542 -0
  944. vllm/model_executor/models/aimv2.py +246 -0
  945. vllm/model_executor/models/apertus.py +579 -0
  946. vllm/model_executor/models/arcee.py +422 -0
  947. vllm/model_executor/models/arctic.py +558 -0
  948. vllm/model_executor/models/aria.py +650 -0
  949. vllm/model_executor/models/aya_vision.py +468 -0
  950. vllm/model_executor/models/baichuan.py +474 -0
  951. vllm/model_executor/models/bailing_moe.py +642 -0
  952. vllm/model_executor/models/bamba.py +514 -0
  953. vllm/model_executor/models/bert.py +665 -0
  954. vllm/model_executor/models/bert_with_rope.py +687 -0
  955. vllm/model_executor/models/blip.py +339 -0
  956. vllm/model_executor/models/blip2.py +712 -0
  957. vllm/model_executor/models/bloom.py +374 -0
  958. vllm/model_executor/models/chameleon.py +1139 -0
  959. vllm/model_executor/models/chatglm.py +476 -0
  960. vllm/model_executor/models/clip.py +407 -0
  961. vllm/model_executor/models/cohere2_vision.py +481 -0
  962. vllm/model_executor/models/commandr.py +465 -0
  963. vllm/model_executor/models/config.py +445 -0
  964. vllm/model_executor/models/dbrx.py +471 -0
  965. vllm/model_executor/models/deepseek.py +497 -0
  966. vllm/model_executor/models/deepseek_eagle.py +240 -0
  967. vllm/model_executor/models/deepseek_mtp.py +289 -0
  968. vllm/model_executor/models/deepseek_v2.py +1444 -0
  969. vllm/model_executor/models/deepseek_vl2.py +658 -0
  970. vllm/model_executor/models/dots1.py +546 -0
  971. vllm/model_executor/models/dots_ocr.py +873 -0
  972. vllm/model_executor/models/ernie45.py +43 -0
  973. vllm/model_executor/models/ernie45_moe.py +607 -0
  974. vllm/model_executor/models/ernie45_vl.py +1527 -0
  975. vllm/model_executor/models/ernie45_vl_moe.py +727 -0
  976. vllm/model_executor/models/ernie_mtp.py +268 -0
  977. vllm/model_executor/models/exaone.py +550 -0
  978. vllm/model_executor/models/exaone4.py +533 -0
  979. vllm/model_executor/models/fairseq2_llama.py +154 -0
  980. vllm/model_executor/models/falcon.py +509 -0
  981. vllm/model_executor/models/falcon_h1.py +674 -0
  982. vllm/model_executor/models/fuyu.py +399 -0
  983. vllm/model_executor/models/gemma.py +425 -0
  984. vllm/model_executor/models/gemma2.py +422 -0
  985. vllm/model_executor/models/gemma3.py +555 -0
  986. vllm/model_executor/models/gemma3_mm.py +721 -0
  987. vllm/model_executor/models/gemma3n.py +1113 -0
  988. vllm/model_executor/models/gemma3n_mm.py +761 -0
  989. vllm/model_executor/models/glm.py +23 -0
  990. vllm/model_executor/models/glm4.py +304 -0
  991. vllm/model_executor/models/glm4_1v.py +1690 -0
  992. vllm/model_executor/models/glm4_moe.py +727 -0
  993. vllm/model_executor/models/glm4_moe_mtp.py +301 -0
  994. vllm/model_executor/models/glm4v.py +654 -0
  995. vllm/model_executor/models/gpt2.py +380 -0
  996. vllm/model_executor/models/gpt_bigcode.py +344 -0
  997. vllm/model_executor/models/gpt_j.py +339 -0
  998. vllm/model_executor/models/gpt_neox.py +330 -0
  999. vllm/model_executor/models/gpt_oss.py +712 -0
  1000. vllm/model_executor/models/granite.py +489 -0
  1001. vllm/model_executor/models/granite_speech.py +794 -0
  1002. vllm/model_executor/models/granitemoe.py +550 -0
  1003. vllm/model_executor/models/granitemoehybrid.py +614 -0
  1004. vllm/model_executor/models/granitemoeshared.py +332 -0
  1005. vllm/model_executor/models/gritlm.py +262 -0
  1006. vllm/model_executor/models/grok1.py +547 -0
  1007. vllm/model_executor/models/h2ovl.py +536 -0
  1008. vllm/model_executor/models/hunyuan_v1.py +1042 -0
  1009. vllm/model_executor/models/hyperclovax_vision.py +1192 -0
  1010. vllm/model_executor/models/idefics2_vision_model.py +417 -0
  1011. vllm/model_executor/models/idefics3.py +756 -0
  1012. vllm/model_executor/models/interfaces.py +959 -0
  1013. vllm/model_executor/models/interfaces_base.py +192 -0
  1014. vllm/model_executor/models/intern_vit.py +441 -0
  1015. vllm/model_executor/models/internlm2.py +450 -0
  1016. vllm/model_executor/models/internlm2_ve.py +148 -0
  1017. vllm/model_executor/models/interns1.py +838 -0
  1018. vllm/model_executor/models/interns1_vit.py +418 -0
  1019. vllm/model_executor/models/internvl.py +1423 -0
  1020. vllm/model_executor/models/jais.py +373 -0
  1021. vllm/model_executor/models/jamba.py +591 -0
  1022. vllm/model_executor/models/jina_vl.py +144 -0
  1023. vllm/model_executor/models/keye.py +1680 -0
  1024. vllm/model_executor/models/keye_vl1_5.py +602 -0
  1025. vllm/model_executor/models/kimi_vl.py +618 -0
  1026. vllm/model_executor/models/lfm2.py +548 -0
  1027. vllm/model_executor/models/llama.py +669 -0
  1028. vllm/model_executor/models/llama4.py +746 -0
  1029. vllm/model_executor/models/llama4_eagle.py +239 -0
  1030. vllm/model_executor/models/llama_eagle.py +179 -0
  1031. vllm/model_executor/models/llama_eagle3.py +296 -0
  1032. vllm/model_executor/models/llava.py +870 -0
  1033. vllm/model_executor/models/llava_next.py +571 -0
  1034. vllm/model_executor/models/llava_next_video.py +476 -0
  1035. vllm/model_executor/models/llava_onevision.py +942 -0
  1036. vllm/model_executor/models/longcat_flash.py +715 -0
  1037. vllm/model_executor/models/longcat_flash_mtp.py +352 -0
  1038. vllm/model_executor/models/mamba.py +275 -0
  1039. vllm/model_executor/models/mamba2.py +291 -0
  1040. vllm/model_executor/models/medusa.py +169 -0
  1041. vllm/model_executor/models/midashenglm.py +792 -0
  1042. vllm/model_executor/models/mimo.py +188 -0
  1043. vllm/model_executor/models/mimo_mtp.py +280 -0
  1044. vllm/model_executor/models/minicpm.py +631 -0
  1045. vllm/model_executor/models/minicpm3.py +230 -0
  1046. vllm/model_executor/models/minicpm_eagle.py +389 -0
  1047. vllm/model_executor/models/minicpmo.py +770 -0
  1048. vllm/model_executor/models/minicpmv.py +1784 -0
  1049. vllm/model_executor/models/minimax_text_01.py +986 -0
  1050. vllm/model_executor/models/minimax_vl_01.py +426 -0
  1051. vllm/model_executor/models/mistral3.py +628 -0
  1052. vllm/model_executor/models/mixtral.py +606 -0
  1053. vllm/model_executor/models/mllama4.py +1076 -0
  1054. vllm/model_executor/models/mlp_speculator.py +206 -0
  1055. vllm/model_executor/models/modernbert.py +374 -0
  1056. vllm/model_executor/models/module_mapping.py +72 -0
  1057. vllm/model_executor/models/molmo.py +1567 -0
  1058. vllm/model_executor/models/moonvit.py +673 -0
  1059. vllm/model_executor/models/motif.py +345 -0
  1060. vllm/model_executor/models/mpt.py +329 -0
  1061. vllm/model_executor/models/nano_nemotron_vl.py +1394 -0
  1062. vllm/model_executor/models/nemotron.py +507 -0
  1063. vllm/model_executor/models/nemotron_h.py +565 -0
  1064. vllm/model_executor/models/nemotron_nas.py +481 -0
  1065. vllm/model_executor/models/nemotron_vl.py +652 -0
  1066. vllm/model_executor/models/nvlm_d.py +203 -0
  1067. vllm/model_executor/models/olmo.py +404 -0
  1068. vllm/model_executor/models/olmo2.py +439 -0
  1069. vllm/model_executor/models/olmoe.py +483 -0
  1070. vllm/model_executor/models/opt.py +412 -0
  1071. vllm/model_executor/models/orion.py +348 -0
  1072. vllm/model_executor/models/ovis.py +559 -0
  1073. vllm/model_executor/models/ovis2_5.py +642 -0
  1074. vllm/model_executor/models/paligemma.py +411 -0
  1075. vllm/model_executor/models/persimmon.py +343 -0
  1076. vllm/model_executor/models/phi.py +356 -0
  1077. vllm/model_executor/models/phi3.py +19 -0
  1078. vllm/model_executor/models/phi3v.py +698 -0
  1079. vllm/model_executor/models/phi4_multimodal.py +1475 -0
  1080. vllm/model_executor/models/phi4mm.py +1279 -0
  1081. vllm/model_executor/models/phi4mm_audio.py +1254 -0
  1082. vllm/model_executor/models/phi4mm_utils.py +1875 -0
  1083. vllm/model_executor/models/phimoe.py +679 -0
  1084. vllm/model_executor/models/pixtral.py +1345 -0
  1085. vllm/model_executor/models/plamo2.py +978 -0
  1086. vllm/model_executor/models/qwen.py +361 -0
  1087. vllm/model_executor/models/qwen2.py +523 -0
  1088. vllm/model_executor/models/qwen2_5_omni_thinker.py +984 -0
  1089. vllm/model_executor/models/qwen2_5_vl.py +1481 -0
  1090. vllm/model_executor/models/qwen2_audio.py +489 -0
  1091. vllm/model_executor/models/qwen2_moe.py +558 -0
  1092. vllm/model_executor/models/qwen2_rm.py +122 -0
  1093. vllm/model_executor/models/qwen2_vl.py +1670 -0
  1094. vllm/model_executor/models/qwen3.py +341 -0
  1095. vllm/model_executor/models/qwen3_moe.py +692 -0
  1096. vllm/model_executor/models/qwen3_next.py +1266 -0
  1097. vllm/model_executor/models/qwen3_next_mtp.py +281 -0
  1098. vllm/model_executor/models/qwen3_vl.py +1613 -0
  1099. vllm/model_executor/models/qwen3_vl_moe.py +358 -0
  1100. vllm/model_executor/models/qwen_vl.py +795 -0
  1101. vllm/model_executor/models/radio.py +576 -0
  1102. vllm/model_executor/models/registry.py +990 -0
  1103. vllm/model_executor/models/roberta.py +252 -0
  1104. vllm/model_executor/models/rvl.py +103 -0
  1105. vllm/model_executor/models/seed_oss.py +485 -0
  1106. vllm/model_executor/models/siglip.py +540 -0
  1107. vllm/model_executor/models/siglip2navit.py +689 -0
  1108. vllm/model_executor/models/skyworkr1v.py +911 -0
  1109. vllm/model_executor/models/smolvlm.py +44 -0
  1110. vllm/model_executor/models/solar.py +504 -0
  1111. vllm/model_executor/models/stablelm.py +341 -0
  1112. vllm/model_executor/models/starcoder2.py +354 -0
  1113. vllm/model_executor/models/step3_text.py +510 -0
  1114. vllm/model_executor/models/step3_vl.py +1072 -0
  1115. vllm/model_executor/models/swin.py +475 -0
  1116. vllm/model_executor/models/tarsier.py +639 -0
  1117. vllm/model_executor/models/telechat2.py +151 -0
  1118. vllm/model_executor/models/teleflm.py +79 -0
  1119. vllm/model_executor/models/terratorch.py +294 -0
  1120. vllm/model_executor/models/transformers.py +948 -0
  1121. vllm/model_executor/models/ultravox.py +654 -0
  1122. vllm/model_executor/models/utils.py +808 -0
  1123. vllm/model_executor/models/vision.py +404 -0
  1124. vllm/model_executor/models/voxtral.py +786 -0
  1125. vllm/model_executor/models/whisper.py +963 -0
  1126. vllm/model_executor/models/zamba2.py +960 -0
  1127. vllm/model_executor/parameter.py +620 -0
  1128. vllm/model_executor/utils.py +86 -0
  1129. vllm/model_executor/warmup/__init__.py +0 -0
  1130. vllm/model_executor/warmup/deep_gemm_warmup.py +230 -0
  1131. vllm/model_executor/warmup/kernel_warmup.py +83 -0
  1132. vllm/multimodal/__init__.py +33 -0
  1133. vllm/multimodal/audio.py +116 -0
  1134. vllm/multimodal/base.py +27 -0
  1135. vllm/multimodal/cache.py +697 -0
  1136. vllm/multimodal/evs.py +273 -0
  1137. vllm/multimodal/hasher.py +102 -0
  1138. vllm/multimodal/image.py +130 -0
  1139. vllm/multimodal/inputs.py +987 -0
  1140. vllm/multimodal/parse.py +511 -0
  1141. vllm/multimodal/processing.py +2148 -0
  1142. vllm/multimodal/profiling.py +284 -0
  1143. vllm/multimodal/registry.py +345 -0
  1144. vllm/multimodal/utils.py +503 -0
  1145. vllm/multimodal/video.py +319 -0
  1146. vllm/outputs.py +324 -0
  1147. vllm/platforms/__init__.py +263 -0
  1148. vllm/platforms/cpu.py +340 -0
  1149. vllm/platforms/cuda.py +668 -0
  1150. vllm/platforms/interface.py +620 -0
  1151. vllm/platforms/rocm.py +497 -0
  1152. vllm/platforms/tpu.py +233 -0
  1153. vllm/platforms/xpu.py +243 -0
  1154. vllm/plugins/__init__.py +72 -0
  1155. vllm/plugins/io_processors/__init__.py +68 -0
  1156. vllm/plugins/io_processors/interface.py +67 -0
  1157. vllm/plugins/lora_resolvers/README.md +16 -0
  1158. vllm/plugins/lora_resolvers/__init__.py +0 -0
  1159. vllm/plugins/lora_resolvers/filesystem_resolver.py +50 -0
  1160. vllm/pooling_params.py +191 -0
  1161. vllm/profiler/__init__.py +0 -0
  1162. vllm/profiler/layerwise_profile.py +375 -0
  1163. vllm/profiler/utils.py +148 -0
  1164. vllm/py.typed +2 -0
  1165. vllm/ray/__init__.py +0 -0
  1166. vllm/ray/lazy_utils.py +22 -0
  1167. vllm/ray/ray_env.py +72 -0
  1168. vllm/reasoning/__init__.py +29 -0
  1169. vllm/reasoning/abs_reasoning_parsers.py +202 -0
  1170. vllm/reasoning/basic_parsers.py +156 -0
  1171. vllm/reasoning/deepseek_r1_reasoning_parser.py +67 -0
  1172. vllm/reasoning/glm4_moe_reasoning_parser.py +151 -0
  1173. vllm/reasoning/gptoss_reasoning_parser.py +87 -0
  1174. vllm/reasoning/granite_reasoning_parser.py +363 -0
  1175. vllm/reasoning/hunyuan_a13b_reasoning_parser.py +245 -0
  1176. vllm/reasoning/mistral_reasoning_parser.py +56 -0
  1177. vllm/reasoning/qwen3_reasoning_parser.py +72 -0
  1178. vllm/reasoning/seedoss_reasoning_parser.py +28 -0
  1179. vllm/reasoning/step3_reasoning_parser.py +109 -0
  1180. vllm/sampling_params.py +593 -0
  1181. vllm/scalar_type.py +349 -0
  1182. vllm/scripts.py +15 -0
  1183. vllm/sequence.py +103 -0
  1184. vllm/tasks.py +11 -0
  1185. vllm/test_utils.py +129 -0
  1186. vllm/third_party/__init__.py +0 -0
  1187. vllm/third_party/pynvml.py +6140 -0
  1188. vllm/tracing.py +136 -0
  1189. vllm/transformers_utils/__init__.py +24 -0
  1190. vllm/transformers_utils/chat_templates/__init__.py +5 -0
  1191. vllm/transformers_utils/chat_templates/registry.py +70 -0
  1192. vllm/transformers_utils/chat_templates/template_basic.jinja +3 -0
  1193. vllm/transformers_utils/chat_templates/template_blip2.jinja +11 -0
  1194. vllm/transformers_utils/chat_templates/template_chatml.jinja +10 -0
  1195. vllm/transformers_utils/chat_templates/template_deepseek_vl2.jinja +23 -0
  1196. vllm/transformers_utils/chat_templates/template_fuyu.jinja +3 -0
  1197. vllm/transformers_utils/chat_templates/template_minicpmv45.jinja +93 -0
  1198. vllm/transformers_utils/config.py +1102 -0
  1199. vllm/transformers_utils/config_parser_base.py +20 -0
  1200. vllm/transformers_utils/configs/__init__.py +63 -0
  1201. vllm/transformers_utils/configs/arctic.py +207 -0
  1202. vllm/transformers_utils/configs/chatglm.py +72 -0
  1203. vllm/transformers_utils/configs/deepseek_v3.py +101 -0
  1204. vllm/transformers_utils/configs/deepseek_vl2.py +216 -0
  1205. vllm/transformers_utils/configs/dotsocr.py +69 -0
  1206. vllm/transformers_utils/configs/eagle.py +84 -0
  1207. vllm/transformers_utils/configs/falcon.py +90 -0
  1208. vllm/transformers_utils/configs/jais.py +237 -0
  1209. vllm/transformers_utils/configs/kimi_vl.py +37 -0
  1210. vllm/transformers_utils/configs/medusa.py +63 -0
  1211. vllm/transformers_utils/configs/midashenglm.py +101 -0
  1212. vllm/transformers_utils/configs/mistral.py +165 -0
  1213. vllm/transformers_utils/configs/mlp_speculator.py +68 -0
  1214. vllm/transformers_utils/configs/moonvit.py +33 -0
  1215. vllm/transformers_utils/configs/nemotron.py +205 -0
  1216. vllm/transformers_utils/configs/nemotron_h.py +259 -0
  1217. vllm/transformers_utils/configs/nemotron_vl.py +56 -0
  1218. vllm/transformers_utils/configs/olmo3.py +80 -0
  1219. vllm/transformers_utils/configs/ovis.py +176 -0
  1220. vllm/transformers_utils/configs/qwen3_next.py +275 -0
  1221. vllm/transformers_utils/configs/radio.py +91 -0
  1222. vllm/transformers_utils/configs/speculators/__init__.py +2 -0
  1223. vllm/transformers_utils/configs/speculators/algos.py +32 -0
  1224. vllm/transformers_utils/configs/speculators/base.py +111 -0
  1225. vllm/transformers_utils/configs/step3_vl.py +123 -0
  1226. vllm/transformers_utils/configs/ultravox.py +116 -0
  1227. vllm/transformers_utils/detokenizer_utils.py +199 -0
  1228. vllm/transformers_utils/dynamic_module.py +60 -0
  1229. vllm/transformers_utils/processor.py +299 -0
  1230. vllm/transformers_utils/processors/__init__.py +16 -0
  1231. vllm/transformers_utils/processors/deepseek_vl2.py +362 -0
  1232. vllm/transformers_utils/processors/ovis.py +420 -0
  1233. vllm/transformers_utils/processors/ovis2_5.py +458 -0
  1234. vllm/transformers_utils/runai_utils.py +104 -0
  1235. vllm/transformers_utils/s3_utils.py +93 -0
  1236. vllm/transformers_utils/tokenizer.py +292 -0
  1237. vllm/transformers_utils/tokenizer_base.py +154 -0
  1238. vllm/transformers_utils/tokenizers/__init__.py +10 -0
  1239. vllm/transformers_utils/tokenizers/mistral.py +521 -0
  1240. vllm/transformers_utils/utils.py +108 -0
  1241. vllm/triton_utils/__init__.py +16 -0
  1242. vllm/triton_utils/importing.py +96 -0
  1243. vllm/usage/__init__.py +0 -0
  1244. vllm/usage/usage_lib.py +259 -0
  1245. vllm/utils/__init__.py +3566 -0
  1246. vllm/utils/deep_gemm.py +319 -0
  1247. vllm/utils/flashinfer.py +443 -0
  1248. vllm/utils/jsontree.py +178 -0
  1249. vllm/utils/tensor_schema.py +235 -0
  1250. vllm/v1/__init__.py +0 -0
  1251. vllm/v1/attention/__init__.py +0 -0
  1252. vllm/v1/attention/backends/__init__.py +0 -0
  1253. vllm/v1/attention/backends/cpu_attn.py +919 -0
  1254. vllm/v1/attention/backends/flash_attn.py +795 -0
  1255. vllm/v1/attention/backends/flashinfer.py +1181 -0
  1256. vllm/v1/attention/backends/flex_attention.py +861 -0
  1257. vllm/v1/attention/backends/gdn_attn.py +332 -0
  1258. vllm/v1/attention/backends/linear_attn.py +67 -0
  1259. vllm/v1/attention/backends/mamba1_attn.py +81 -0
  1260. vllm/v1/attention/backends/mamba2_attn.py +232 -0
  1261. vllm/v1/attention/backends/mamba_attn.py +52 -0
  1262. vllm/v1/attention/backends/mla/__init__.py +0 -0
  1263. vllm/v1/attention/backends/mla/common.py +1783 -0
  1264. vllm/v1/attention/backends/mla/cutlass_mla.py +248 -0
  1265. vllm/v1/attention/backends/mla/flashattn_mla.py +271 -0
  1266. vllm/v1/attention/backends/mla/flashinfer_mla.py +114 -0
  1267. vllm/v1/attention/backends/mla/flashmla.py +203 -0
  1268. vllm/v1/attention/backends/mla/flashmla_sparse.py +544 -0
  1269. vllm/v1/attention/backends/mla/indexer.py +342 -0
  1270. vllm/v1/attention/backends/mla/rocm_aiter_mla.py +255 -0
  1271. vllm/v1/attention/backends/mla/triton_mla.py +177 -0
  1272. vllm/v1/attention/backends/pallas.py +409 -0
  1273. vllm/v1/attention/backends/rocm_aiter_fa.py +549 -0
  1274. vllm/v1/attention/backends/rocm_attn.py +426 -0
  1275. vllm/v1/attention/backends/short_conv_attn.py +94 -0
  1276. vllm/v1/attention/backends/tree_attn.py +451 -0
  1277. vllm/v1/attention/backends/triton_attn.py +361 -0
  1278. vllm/v1/attention/backends/utils.py +990 -0
  1279. vllm/v1/attention/backends/xformers.py +438 -0
  1280. vllm/v1/core/__init__.py +0 -0
  1281. vllm/v1/core/block_pool.py +416 -0
  1282. vllm/v1/core/encoder_cache_manager.py +333 -0
  1283. vllm/v1/core/kv_cache_coordinator.py +440 -0
  1284. vllm/v1/core/kv_cache_manager.py +399 -0
  1285. vllm/v1/core/kv_cache_utils.py +1291 -0
  1286. vllm/v1/core/sched/__init__.py +0 -0
  1287. vllm/v1/core/sched/async_scheduler.py +47 -0
  1288. vllm/v1/core/sched/interface.py +158 -0
  1289. vllm/v1/core/sched/output.py +166 -0
  1290. vllm/v1/core/sched/request_queue.py +224 -0
  1291. vllm/v1/core/sched/scheduler.py +1296 -0
  1292. vllm/v1/core/sched/utils.py +69 -0
  1293. vllm/v1/core/single_type_kv_cache_manager.py +671 -0
  1294. vllm/v1/cudagraph_dispatcher.py +125 -0
  1295. vllm/v1/engine/__init__.py +203 -0
  1296. vllm/v1/engine/async_llm.py +742 -0
  1297. vllm/v1/engine/coordinator.py +357 -0
  1298. vllm/v1/engine/core.py +1235 -0
  1299. vllm/v1/engine/core_client.py +1334 -0
  1300. vllm/v1/engine/detokenizer.py +349 -0
  1301. vllm/v1/engine/exceptions.py +17 -0
  1302. vllm/v1/engine/llm_engine.py +370 -0
  1303. vllm/v1/engine/logprobs.py +201 -0
  1304. vllm/v1/engine/output_processor.py +576 -0
  1305. vllm/v1/engine/parallel_sampling.py +133 -0
  1306. vllm/v1/engine/processor.py +545 -0
  1307. vllm/v1/engine/utils.py +860 -0
  1308. vllm/v1/executor/__init__.py +0 -0
  1309. vllm/v1/executor/abstract.py +137 -0
  1310. vllm/v1/executor/multiproc_executor.py +726 -0
  1311. vllm/v1/executor/ray_distributed_executor.py +108 -0
  1312. vllm/v1/executor/utils.py +23 -0
  1313. vllm/v1/kv_cache_interface.py +375 -0
  1314. vllm/v1/kv_offload/__init__.py +0 -0
  1315. vllm/v1/kv_offload/abstract.py +165 -0
  1316. vllm/v1/kv_offload/backend.py +96 -0
  1317. vllm/v1/kv_offload/backends/__init__.py +0 -0
  1318. vllm/v1/kv_offload/backends/cpu.py +61 -0
  1319. vllm/v1/kv_offload/cpu.py +75 -0
  1320. vllm/v1/kv_offload/factory.py +56 -0
  1321. vllm/v1/kv_offload/lru_manager.py +132 -0
  1322. vllm/v1/kv_offload/mediums.py +39 -0
  1323. vllm/v1/kv_offload/spec.py +61 -0
  1324. vllm/v1/kv_offload/worker/__init__.py +0 -0
  1325. vllm/v1/kv_offload/worker/cpu_gpu.py +171 -0
  1326. vllm/v1/kv_offload/worker/worker.py +142 -0
  1327. vllm/v1/metrics/__init__.py +0 -0
  1328. vllm/v1/metrics/loggers.py +741 -0
  1329. vllm/v1/metrics/prometheus.py +82 -0
  1330. vllm/v1/metrics/ray_wrappers.py +152 -0
  1331. vllm/v1/metrics/reader.py +246 -0
  1332. vllm/v1/metrics/stats.py +257 -0
  1333. vllm/v1/outputs.py +161 -0
  1334. vllm/v1/pool/__init__.py +0 -0
  1335. vllm/v1/pool/metadata.py +77 -0
  1336. vllm/v1/request.py +241 -0
  1337. vllm/v1/sample/__init__.py +0 -0
  1338. vllm/v1/sample/logits_processor/__init__.py +294 -0
  1339. vllm/v1/sample/logits_processor/builtin.py +275 -0
  1340. vllm/v1/sample/logits_processor/interface.py +97 -0
  1341. vllm/v1/sample/logits_processor/state.py +161 -0
  1342. vllm/v1/sample/metadata.py +43 -0
  1343. vllm/v1/sample/ops/__init__.py +0 -0
  1344. vllm/v1/sample/ops/bad_words.py +39 -0
  1345. vllm/v1/sample/ops/logprobs.py +26 -0
  1346. vllm/v1/sample/ops/penalties.py +43 -0
  1347. vllm/v1/sample/ops/topk_topp_sampler.py +292 -0
  1348. vllm/v1/sample/rejection_sampler.py +623 -0
  1349. vllm/v1/sample/sampler.py +285 -0
  1350. vllm/v1/sample/tpu/__init__.py +0 -0
  1351. vllm/v1/sample/tpu/metadata.py +124 -0
  1352. vllm/v1/sample/tpu/sampler.py +213 -0
  1353. vllm/v1/serial_utils.py +423 -0
  1354. vllm/v1/spec_decode/__init__.py +0 -0
  1355. vllm/v1/spec_decode/eagle.py +1011 -0
  1356. vllm/v1/spec_decode/medusa.py +66 -0
  1357. vllm/v1/spec_decode/metadata.py +62 -0
  1358. vllm/v1/spec_decode/metrics.py +211 -0
  1359. vllm/v1/spec_decode/ngram_proposer.py +276 -0
  1360. vllm/v1/spec_decode/utils.py +14 -0
  1361. vllm/v1/structured_output/__init__.py +295 -0
  1362. vllm/v1/structured_output/backend_guidance.py +245 -0
  1363. vllm/v1/structured_output/backend_lm_format_enforcer.py +167 -0
  1364. vllm/v1/structured_output/backend_outlines.py +320 -0
  1365. vllm/v1/structured_output/backend_types.py +134 -0
  1366. vllm/v1/structured_output/backend_xgrammar.py +327 -0
  1367. vllm/v1/structured_output/request.py +86 -0
  1368. vllm/v1/structured_output/utils.py +454 -0
  1369. vllm/v1/utils.py +396 -0
  1370. vllm/v1/worker/__init__.py +0 -0
  1371. vllm/v1/worker/block_table.py +210 -0
  1372. vllm/v1/worker/cpu_model_runner.py +175 -0
  1373. vllm/v1/worker/cpu_worker.py +156 -0
  1374. vllm/v1/worker/gpu_input_batch.py +863 -0
  1375. vllm/v1/worker/gpu_model_runner.py +4160 -0
  1376. vllm/v1/worker/gpu_ubatch_wrapper.py +399 -0
  1377. vllm/v1/worker/gpu_worker.py +710 -0
  1378. vllm/v1/worker/kv_connector_model_runner_mixin.py +132 -0
  1379. vllm/v1/worker/lora_model_runner_mixin.py +183 -0
  1380. vllm/v1/worker/tpu_input_batch.py +587 -0
  1381. vllm/v1/worker/tpu_model_runner.py +1946 -0
  1382. vllm/v1/worker/tpu_worker.py +346 -0
  1383. vllm/v1/worker/ubatch_splitting.py +192 -0
  1384. vllm/v1/worker/ubatch_utils.py +27 -0
  1385. vllm/v1/worker/ubatching.py +224 -0
  1386. vllm/v1/worker/utils.py +344 -0
  1387. vllm/v1/worker/worker_base.py +65 -0
  1388. vllm/v1/worker/xpu_model_runner.py +57 -0
  1389. vllm/v1/worker/xpu_worker.py +179 -0
  1390. vllm/version.py +41 -0
  1391. vllm/vllm_flash_attn/.gitkeep +0 -0
  1392. vllm/worker/__init__.py +0 -0
  1393. vllm/worker/worker_base.py +279 -0
  1394. vllm_cpu-0.11.0.post2.dist-info/METADATA +348 -0
  1395. vllm_cpu-0.11.0.post2.dist-info/RECORD +1398 -0
  1396. vllm_cpu-0.11.0.post2.dist-info/WHEEL +5 -0
  1397. vllm_cpu-0.11.0.post2.dist-info/entry_points.txt +5 -0
  1398. vllm_cpu-0.11.0.post2.dist-info/top_level.txt +1 -0
@@ -0,0 +1,1784 @@
1
+ # SPDX-License-Identifier: Apache-2.0
2
+ # SPDX-FileCopyrightText: Copyright contributors to the vLLM project
3
+
4
+ # Adapted from
5
+ # https://github.com/huggingface/transformers/blob/v4.28.0/src/transformers/models/llama/modeling_llama.py
6
+ # Copyright 2023 The vLLM team.
7
+ # Copyright 2022 EleutherAI and the HuggingFace Inc. team. All rights reserved.
8
+ #
9
+ # This code is based on EleutherAI's GPT-NeoX library and the GPT-NeoX
10
+ # and OPT implementations in this library. It has been modified from its
11
+ # original forms to accommodate minor architectural differences compared
12
+ # to GPT-NeoX and OPT used by the Meta AI team that trained the model.
13
+ #
14
+ # Licensed under the Apache License, Version 2.0 (the "License");
15
+ # you may not use this file except in compliance with the License.
16
+ # You may obtain a copy of the License at
17
+ #
18
+ # http://www.apache.org/licenses/LICENSE-2.0
19
+ #
20
+ # Unless required by applicable law or agreed to in writing, software
21
+ # distributed under the License is distributed on an "AS IS" BASIS,
22
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
23
+ # See the License for the specific language governing permissions and
24
+ # limitations under the License.
25
+ """Inference-only MiniCPM-V model compatible with HuggingFace weights."""
26
+ import math
27
+ from collections import defaultdict
28
+ from collections.abc import Iterable, Mapping, Sequence
29
+ from functools import partial
30
+ from itertools import chain
31
+ from typing import Annotated, Any, Callable, Literal, Optional, Union
32
+
33
+ import numpy as np
34
+ import torch
35
+ import torch.types
36
+ from torch import nn
37
+ from torch.nn.init import trunc_normal_
38
+ from transformers import BatchFeature, PretrainedConfig
39
+ from typing_extensions import TypeVar
40
+
41
+ from vllm.config import VllmConfig
42
+ from vllm.model_executor.layers.quantization import QuantizationConfig
43
+ from vllm.model_executor.layers.quantization.awq import AWQConfig
44
+ from vllm.model_executor.layers.quantization.awq_marlin import AWQMarlinConfig
45
+ from vllm.model_executor.layers.resampler import (BaseResampler, Resampler2,
46
+ get_2d_sincos_pos_embed)
47
+ from vllm.model_executor.model_loader.utils import set_default_torch_dtype
48
+ from vllm.model_executor.models.llama import LlamaForCausalLM
49
+ from vllm.model_executor.models.minicpm import MiniCPMForCausalLM
50
+ from vllm.model_executor.models.module_mapping import MultiModelKeys
51
+ from vllm.model_executor.models.qwen2 import Qwen2ForCausalLM
52
+ from vllm.model_executor.models.qwen3 import Qwen3ForCausalLM
53
+ from vllm.multimodal import MULTIMODAL_REGISTRY
54
+ from vllm.multimodal.inputs import (MultiModalDataDict, MultiModalFieldConfig,
55
+ MultiModalKwargsItems, NestedTensors)
56
+ from vllm.multimodal.parse import (DictEmbeddingItems, ImageItem,
57
+ ImageProcessorItems, ImageSize,
58
+ ModalityData, ModalityDataItems,
59
+ MultiModalDataItems, MultiModalDataParser,
60
+ VideoItem, VideoProcessorItems)
61
+ from vllm.multimodal.processing import (BaseMultiModalProcessor,
62
+ BaseProcessingInfo, PromptReplacement,
63
+ PromptUpdate, PromptUpdateDetails,
64
+ ResolvedPromptUpdate, _seq2text)
65
+ from vllm.multimodal.profiling import BaseDummyInputsBuilder
66
+ from vllm.platforms import current_platform
67
+ from vllm.sequence import IntermediateTensors
68
+ from vllm.utils import flatten_2d_lists
69
+ from vllm.utils.tensor_schema import TensorSchema, TensorShape
70
+
71
+ from .idefics2_vision_model import Idefics2VisionTransformer
72
+ from .interfaces import (MultiModalEmbeddings, SupportsLoRA,
73
+ SupportsMultiModal, SupportsPP)
74
+ from .utils import (AutoWeightsLoader, flatten_bn, maybe_prefix,
75
+ merge_multimodal_embeddings)
76
+
77
+ # For profile run
78
+ _MAX_FRAMES_PER_VIDEO = 16
79
+
80
+
81
+ class MiniCPMVImagePixelInputs(TensorSchema):
82
+ """
83
+ Dimensions:
84
+ - bns: Batch size * number of images * number of slices
85
+ - bn: Batch size * number of images
86
+ - c: Number of channels
87
+ - h: Height
88
+ - w: Width
89
+ """
90
+
91
+ type: Literal["pixel_values"] = "pixel_values"
92
+
93
+ # Note that the image size may vary, so we pass it as a list instead of a
94
+ # batched tensor.
95
+ pixel_values: Annotated[
96
+ list[torch.Tensor],
97
+ TensorShape("bns", "c", "h", "w", dynamic_dims={"h", "w"}),
98
+ ]
99
+ tgt_sizes: Annotated[
100
+ torch.Tensor,
101
+ TensorShape("bns", 2), # This should be in `(height, width)` format.
102
+ ]
103
+ num_slices: Annotated[
104
+ torch.Tensor,
105
+ TensorShape("bn"),
106
+ ]
107
+
108
+
109
+ class MiniCPMVImageEmbeddingInputs(TensorSchema):
110
+ """
111
+ Dimensions:
112
+ - bn: Batch size * number of images
113
+ - ns: Number of slices
114
+ - hs: Hidden size (must match language model backbone)
115
+ """
116
+
117
+ type: Literal["image_embeds"]
118
+ image_embeds: Annotated[
119
+ Union[torch.Tensor, list[torch.Tensor]],
120
+ TensorShape("bn", "ns", "hs"),
121
+ ]
122
+
123
+
124
+ MiniCPMVImageInputs = Union[MiniCPMVImagePixelInputs,
125
+ MiniCPMVImageEmbeddingInputs]
126
+
127
+ DEFAULT_LN = partial(nn.LayerNorm, eps=1e-6)
128
+
129
+
130
+ class Resampler2_5(BaseResampler):
131
+
132
+ def __init__(self,
133
+ num_queries: int,
134
+ embed_dim: int,
135
+ num_heads: int,
136
+ kv_dim: Optional[int] = None,
137
+ norm_layer: Callable[[int], nn.LayerNorm] = DEFAULT_LN,
138
+ max_size: tuple[int, int] = (70, 70),
139
+ quant_config: Optional[QuantizationConfig] = None,
140
+ prefix: str = "") -> None:
141
+ super().__init__(num_queries,
142
+ embed_dim,
143
+ num_heads,
144
+ kv_dim,
145
+ norm_layer,
146
+ quant_config=quant_config,
147
+ prefix=prefix)
148
+
149
+ self.max_size = max_size
150
+ self._set_2d_pos_cache(self.max_size)
151
+
152
+ def _set_2d_pos_cache(self,
153
+ max_size: tuple[int, int],
154
+ device: torch.types.Device = "cpu") -> None:
155
+ pos_embed_arr = get_2d_sincos_pos_embed(self.embed_dim,
156
+ max_size,
157
+ version=(2, 5))
158
+ pos_embed = torch.from_numpy(pos_embed_arr).float().to(device)
159
+ self.register_buffer("pos_embed", pos_embed, persistent=False)
160
+
161
+ def _adjust_pos_cache(self, tgt_sizes: torch.Tensor,
162
+ device: torch.types.Device) -> None:
163
+ max_h = tgt_sizes[:, 0].max().item()
164
+ max_w = tgt_sizes[:, 1].max().item()
165
+ assert isinstance(max_h, int) and isinstance(max_w, int)
166
+
167
+ if max_h > self.max_size[0] or max_w > self.max_size[1]:
168
+ self.max_size = (
169
+ max(max_h, self.max_size[0]),
170
+ max(max_w, self.max_size[1]),
171
+ )
172
+ self._set_2d_pos_cache(self.max_size, device)
173
+
174
+ def forward(self, x: torch.Tensor,
175
+ tgt_sizes: torch.Tensor) -> torch.Tensor:
176
+ assert x.shape[0] == tgt_sizes.shape[0]
177
+ bs = x.shape[0]
178
+
179
+ device = x.device
180
+ dtype = x.dtype
181
+
182
+ patch_len = tgt_sizes[:, 0] * tgt_sizes[:, 1]
183
+
184
+ self._adjust_pos_cache(tgt_sizes, device=device)
185
+
186
+ max_patch_len = patch_len.max().item()
187
+ assert isinstance(max_patch_len, int)
188
+
189
+ key_padding_mask = torch.zeros((bs, max_patch_len),
190
+ dtype=torch.bool,
191
+ device=device)
192
+
193
+ pos_embed = []
194
+ for i in range(bs):
195
+ tgt_h, tgt_w = tgt_sizes[i].tolist()
196
+ pos_embed.append(self.pos_embed[:tgt_h, :tgt_w, :].reshape(
197
+ (tgt_h * tgt_w, -1)).to(dtype)) # patches * D
198
+ key_padding_mask[i, patch_len[i]:] = True
199
+ pos_embed = torch.nn.utils.rnn.pad_sequence(pos_embed,
200
+ batch_first=True,
201
+ padding_value=0.0).permute(
202
+ 1, 0,
203
+ 2) # BLD => L * B * D
204
+ x, _ = self.kv_proj(x) # B * L * D
205
+ x = self.ln_kv(x).permute(1, 0, 2) # L * B * D
206
+
207
+ q = self.ln_q(self.query) # Q * D
208
+
209
+ out = self.attn(
210
+ self._repeat(q, bs), # Q * B * D
211
+ x + pos_embed, # L * B * D + L * B * D
212
+ x,
213
+ key_padding_mask=key_padding_mask,
214
+ )[0]
215
+ # out: Q * B * D
216
+ x = out.permute(1, 0, 2) # B * Q * D
217
+
218
+ x = self.ln_post(x)
219
+ x = x @ self.proj
220
+ return x
221
+
222
+
223
+ class Resampler4_5(Resampler2_5):
224
+
225
+ def __init__(self,
226
+ num_queries: int,
227
+ embed_dim: int,
228
+ num_heads: int,
229
+ kv_dim: Optional[int] = None,
230
+ norm_layer: Callable[[int], nn.LayerNorm] = DEFAULT_LN,
231
+ max_size: tuple[int, int] = (70, 70),
232
+ max_temporal_size: int = 36000,
233
+ quant_config: Optional[QuantizationConfig] = None,
234
+ prefix: str = "") -> None:
235
+ super().__init__(num_queries,
236
+ embed_dim,
237
+ num_heads,
238
+ kv_dim,
239
+ norm_layer,
240
+ max_size,
241
+ quant_config=quant_config,
242
+ prefix=prefix)
243
+
244
+ trunc_normal_(self.query, std=.02)
245
+ self.max_temporal_size = max_temporal_size
246
+ self._set_temporal_pos_cache(self.max_temporal_size)
247
+ self.apply(self._init_weights)
248
+
249
+ def get_1d_sincos_pos_embed_from_temporal_size(self, embed_dim: int,
250
+ pos: np.ndarray):
251
+ """
252
+ embed_dim: output dimension for each position
253
+ pos: a list of positions to be encoded: size (M,)
254
+ out: (M, D)
255
+ """
256
+ assert embed_dim % 2 == 0
257
+ omega = np.arange(embed_dim // 2, dtype=np.float32)
258
+ omega /= embed_dim / 2.
259
+ omega = 1. / 10000**omega # (D/2,)
260
+
261
+ pos = pos.reshape(-1) # (M,)
262
+ out = np.einsum('m,d->md', pos, omega) # (M, D/2), outer product
263
+
264
+ emb_sin = np.sin(out) # (M, D/2)
265
+ emb_cos = np.cos(out) # (M, D/2)
266
+
267
+ emb = np.concatenate([emb_sin, emb_cos], axis=1) # (M, D)
268
+ return emb
269
+
270
+ def _set_temporal_pos_cache(self,
271
+ max_temporal_size: int,
272
+ device: torch.types.Device = "cpu") -> None:
273
+ temporal_size = np.arange(max_temporal_size, dtype=np.float32)
274
+ pos_embed = torch.from_numpy(
275
+ self.get_1d_sincos_pos_embed_from_temporal_size(
276
+ self.embed_dim, temporal_size)).float().to(device)
277
+ self.register_buffer("temporal_pos_embed", pos_embed, persistent=False)
278
+
279
+ def _adjust_temporal_pos_cache(self,
280
+ max_temporal_size: int,
281
+ device: torch.types.Device = "cpu"):
282
+ if max_temporal_size > self.max_temporal_size:
283
+ self.max_temporal_size = max_temporal_size
284
+ self._set_temporal_pos_cache(self.max_temporal_size, device)
285
+
286
+ def _init_weights(self, m: Union[nn.Linear, nn.LayerNorm]):
287
+ if isinstance(m, nn.Linear):
288
+ trunc_normal_(m.weight, std=.02)
289
+ if isinstance(m, nn.Linear) and m.bias is not None:
290
+ nn.init.constant_(m.bias, 0)
291
+ elif isinstance(m, nn.LayerNorm):
292
+ nn.init.constant_(m.bias, 0)
293
+ nn.init.constant_(m.weight, 1.0)
294
+
295
+ def forward(
296
+ self,
297
+ x: torch.Tensor,
298
+ tgt_sizes: torch.Tensor,
299
+ # temporal_ids for high refresh rate videos
300
+ temporal_ids=None
301
+ ) -> torch.Tensor:
302
+ assert x.shape[0] == tgt_sizes.shape[0]
303
+ bs = x.shape[0]
304
+
305
+ device = x.device
306
+ dtype = x.dtype
307
+
308
+ patch_len = tgt_sizes[:, 0] * tgt_sizes[:, 1]
309
+
310
+ self._adjust_pos_cache(tgt_sizes, device=device)
311
+
312
+ temporal_pos_emb = False
313
+ temporal_ids_flatten = None
314
+ if temporal_ids is not None:
315
+ # example: [[-1], [-1], [2, 6, 9]]
316
+ temporal_ids_flatten = list(chain.from_iterable(temporal_ids))
317
+ max_temporal_size = max(temporal_ids_flatten, default=0)
318
+ if max_temporal_size > -1:
319
+ temporal_pos_emb = True
320
+ if max_temporal_size > self.max_temporal_size:
321
+ self._adjust_temporal_pos_cache(max_temporal_size, device)
322
+
323
+ max_patch_len = patch_len.max().item()
324
+ assert isinstance(max_patch_len, int)
325
+
326
+ key_padding_mask = torch.zeros((bs, max_patch_len),
327
+ dtype=torch.bool,
328
+ device=device)
329
+
330
+ x, _ = self.kv_proj(x) # B * L * D
331
+ x = self.ln_kv(x).permute(1, 0, 2) # L * B * D
332
+ q = self.ln_q(self.query) # Q * D
333
+
334
+ pos_embed_2d = []
335
+ pos_embed_temporal = []
336
+ for i in range(bs):
337
+ tgt_h, tgt_w = tgt_sizes[i]
338
+ if temporal_pos_emb:
339
+ if temporal_ids_flatten[i] == -1:
340
+ pos_embed_temporal.append(
341
+ torch.zeros(self.embed_dim, dtype=dtype,
342
+ device=device))
343
+ else:
344
+ pos_embed_temporal.append(self.temporal_pos_embed[
345
+ temporal_ids_flatten[i]].to(dtype)) # D
346
+
347
+ pos_embed_2d.append(self.pos_embed[:tgt_h, :tgt_w, :].reshape(
348
+ (tgt_h * tgt_w, -1)).to(dtype)) # patches * D
349
+ key_padding_mask[i, patch_len[i]:] = True
350
+
351
+ pos_embed_2d = torch.nn.utils.rnn.pad_sequence(
352
+ pos_embed_2d, batch_first=True,
353
+ padding_value=0.0).permute(1, 0, 2) # BLD => L * B * D
354
+
355
+ k = x
356
+ v = x + pos_embed_2d
357
+ if pos_embed_temporal:
358
+ k += torch.stack(pos_embed_temporal, dim=0)
359
+ bs = len(temporal_ids)
360
+ merge_k = []
361
+ merge_v = []
362
+ merge_key_padding_mask = []
363
+
364
+ start = 0
365
+ for tp in temporal_ids:
366
+ end = start + len(tp)
367
+ # L * (end-start) * D -> (end-start) * L * D
368
+ # -> 1 * L*(end-start) * D
369
+ merge_k.append(k[:, start:end, :].permute(1, 0, 2).reshape(
370
+ -1, self.embed_dim))
371
+ merge_v.append(v[:, start:end, :].permute(1, 0, 2).reshape(
372
+ -1, self.embed_dim))
373
+ merge_key_padding_mask.append(
374
+ key_padding_mask[start:end, :].reshape(-1, 1))
375
+
376
+ start = end
377
+
378
+ k = torch.nn.utils.rnn.pad_sequence(merge_k,
379
+ batch_first=True,
380
+ padding_value=0.0).permute(
381
+ 1, 0, 2) # L*(end-start)
382
+ v = torch.nn.utils.rnn.pad_sequence(merge_v,
383
+ batch_first=True,
384
+ padding_value=0.0).permute(
385
+ 1, 0, 2) # L*(end-start)
386
+ key_padding_mask = torch.nn.utils.rnn.pad_sequence(
387
+ merge_key_padding_mask, batch_first=True,
388
+ padding_value=True).squeeze(-1)
389
+
390
+ out = self.attn(
391
+ self._repeat(q, bs), # Q * B * D
392
+ k, # L * B * D + L * B * D
393
+ v,
394
+ key_padding_mask=key_padding_mask,
395
+ )[0]
396
+ # out: Q * B * D
397
+ x = out.permute(1, 0, 2) # B * Q * D
398
+
399
+ x = self.ln_post(x)
400
+ x = x @ self.proj
401
+ return x
402
+
403
+
404
+ def get_version_by_config(config: PretrainedConfig) -> tuple[int, ...]:
405
+ version_float = getattr(config, "version", None)
406
+
407
+ # The old configs do not include version number
408
+ # TODO: Remove this after the HF repos are updated
409
+ if version_float is None:
410
+ if config.hidden_size == 2304 and config.query_num == 64:
411
+ return (2, 0)
412
+ return (2, 5)
413
+ version_str = str(version_float)
414
+ return tuple(int(x) for x in version_str.split("."))
415
+
416
+
417
+ def _minicpmv_field_config(hf_inputs: Mapping[str, torch.Tensor]):
418
+ pixel_values = hf_inputs.get("pixel_values", torch.empty(0))
419
+ num_images = len(pixel_values)
420
+
421
+ video_pixel_values = hf_inputs.get("video_pixel_values", torch.empty(0))
422
+ num_videos = len(video_pixel_values)
423
+
424
+ return dict(
425
+ pixel_values=MultiModalFieldConfig.batched("image"),
426
+ image_sizes=MultiModalFieldConfig.batched("image"),
427
+ tgt_sizes=MultiModalFieldConfig.batched("image"),
428
+ image_embeds=MultiModalFieldConfig.batched("image"),
429
+ video_pixel_values=MultiModalFieldConfig.batched("video"),
430
+ video_image_sizes=MultiModalFieldConfig.batched("video"),
431
+ video_tgt_sizes=MultiModalFieldConfig.batched("video"),
432
+ video_embeds=MultiModalFieldConfig.batched("video"),
433
+ image_token_id=MultiModalFieldConfig.shared("image", num_images),
434
+ video_token_id=MultiModalFieldConfig.shared("video", num_videos),
435
+ )
436
+
437
+
438
+ class MiniCPMVImageEmbeddingItems(DictEmbeddingItems):
439
+
440
+ def __init__(
441
+ self,
442
+ data: Mapping[str, torch.Tensor],
443
+ fields_factory: Callable[
444
+ [Mapping[str, torch.Tensor]],
445
+ Mapping[str, MultiModalFieldConfig],
446
+ ],
447
+ ) -> None:
448
+ super().__init__(
449
+ data,
450
+ modality="image",
451
+ required_fields={"image_embeds", "image_sizes"},
452
+ fields_factory=fields_factory,
453
+ )
454
+
455
+ def get_image_size(self, index: int) -> ImageSize:
456
+ image_size = self.get(index)["image_sizes"].tolist()
457
+ return ImageSize(width=image_size[0], height=image_size[1])
458
+
459
+
460
+ class MiniCPMVVideoEmbeddingItems(DictEmbeddingItems):
461
+
462
+ def __init__(
463
+ self,
464
+ data: Mapping[str, torch.Tensor],
465
+ fields_factory: Callable[
466
+ [Mapping[str, torch.Tensor]],
467
+ Mapping[str, MultiModalFieldConfig],
468
+ ],
469
+ ) -> None:
470
+ super().__init__(
471
+ data,
472
+ modality="video",
473
+ required_fields={"video_embeds", "video_image_sizes"},
474
+ fields_factory=fields_factory,
475
+ )
476
+
477
+ def get_frame_size(self, index: int) -> ImageSize:
478
+ frame_size = self.get(index)["video_image_sizes"].tolist()
479
+ return ImageSize(width=frame_size[0], height=frame_size[1])
480
+
481
+ def get_num_frames(self, index: int) -> int:
482
+ return len(self.get(index)["video_image_sizes"])
483
+
484
+
485
+ class MiniCPMVMultiModalDataParser(MultiModalDataParser):
486
+
487
+ def _parse_image_data(
488
+ self,
489
+ data: Union[dict[str, torch.Tensor], ModalityData[ImageItem]],
490
+ ) -> Optional[ModalityDataItems[Any, Any]]:
491
+ if isinstance(data, dict):
492
+ return MiniCPMVImageEmbeddingItems(
493
+ data,
494
+ fields_factory=_minicpmv_field_config,
495
+ )
496
+
497
+ return super()._parse_image_data(data)
498
+
499
+ def _parse_video_data(
500
+ self,
501
+ data: Union[dict[str, torch.Tensor], ModalityData[VideoItem]],
502
+ ) -> Optional[ModalityDataItems[Any, Any]]:
503
+ if isinstance(data, dict):
504
+ return MiniCPMVVideoEmbeddingItems(
505
+ data,
506
+ fields_factory=_minicpmv_field_config,
507
+ )
508
+
509
+ return super()._parse_video_data(data)
510
+
511
+
512
+ class MiniCPMVProcessingInfo(BaseProcessingInfo):
513
+ image_pattern = "(<image>./</image>)"
514
+ video_pattern = "(<video>./</video>)"
515
+
516
+ def get_hf_config(self):
517
+ return self.ctx.get_hf_config()
518
+
519
+ def get_hf_processor(self, **kwargs: object):
520
+ hf_processor = self.ctx.get_hf_processor(**kwargs)
521
+
522
+ # NumPy arrays are considered as Iterable but not Sequence in
523
+ # https://github.com/huggingface/transformers/blob/main/src/transformers/image_transforms.py#L428
524
+ image_processor = hf_processor.image_processor # type: ignore
525
+ for attr in ("mean", "std"):
526
+ val = getattr(image_processor, attr)
527
+ if isinstance(val, np.ndarray):
528
+ setattr(image_processor, attr, val.tolist())
529
+
530
+ return hf_processor
531
+
532
+ def get_image_processor(self, **kwargs: object):
533
+ return self.get_hf_processor(**kwargs).image_processor
534
+
535
+ def get_model_version(self):
536
+ return get_version_by_config(self.get_hf_config())
537
+
538
+ def get_supported_mm_limits(self) -> Mapping[str, Optional[int]]:
539
+ mm_limits = {"image": None}
540
+ if self.get_model_version() in {(2, 6), (4, 0), (4, 5)}:
541
+ mm_limits["video"] = None
542
+
543
+ return mm_limits
544
+
545
+ def get_slice_image_placeholder(
546
+ self,
547
+ image_size: ImageSize,
548
+ # For MiniCPM V/O 2.6
549
+ image_idx: int = 0,
550
+ max_slice_nums: Optional[int] = None,
551
+ use_image_id: bool = True,
552
+ ) -> str:
553
+ image_processor = self.get_image_processor()
554
+ version = self.get_model_version()
555
+
556
+ if version == (2, 0) or version == (2, 5):
557
+ return image_processor.get_slice_image_placeholder(image_size)
558
+
559
+ return image_processor.get_slice_image_placeholder(
560
+ image_size,
561
+ image_idx=image_idx,
562
+ max_slice_nums=max_slice_nums,
563
+ use_image_id=use_image_id,
564
+ )
565
+
566
+ def get_sliced_grid(
567
+ self,
568
+ image_size: ImageSize,
569
+ # For MiniCPM V/O 2.6
570
+ max_slice_nums: Optional[int] = None,
571
+ ) -> Optional[tuple[int, int]]:
572
+ image_processor = self.get_image_processor()
573
+ version = self.get_model_version()
574
+
575
+ if version == (2, 0) or version == (2, 5):
576
+ return image_processor.get_sliced_grid(image_size)
577
+
578
+ if max_slice_nums is None:
579
+ max_slice_nums = image_processor.max_slice_nums
580
+
581
+ return image_processor.get_sliced_grid(
582
+ image_size,
583
+ max_slice_nums=max_slice_nums,
584
+ )
585
+
586
+ def get_num_image_tokens(
587
+ self,
588
+ image_size: ImageSize,
589
+ max_slice_nums: Optional[int] = None,
590
+ ) -> int:
591
+ image_processor = self.get_image_processor()
592
+
593
+ grid = self.get_sliced_grid(
594
+ image_size,
595
+ max_slice_nums=max_slice_nums,
596
+ )
597
+ if grid is None:
598
+ ncols = nrows = 0
599
+ else:
600
+ ncols, nrows = grid
601
+
602
+ return (ncols * nrows + 1) * image_processor.image_feature_size
603
+
604
+ def get_max_image_tokens(self) -> int:
605
+ image_size = self.get_image_size_with_most_features()
606
+ return self.get_num_image_tokens(image_size)
607
+
608
+ def get_image_max_slice_num(self) -> int:
609
+ return getattr(self.get_hf_config(), "max_slice_num", 9)
610
+
611
+ def get_image_size_with_most_features(self) -> ImageSize:
612
+ image_size = getattr(self.get_hf_config(), "image_size", 448)
613
+ max_slice_num = self.get_image_max_slice_num()
614
+ return ImageSize(width=image_size, height=image_size * max_slice_num)
615
+
616
+ def get_max_video_frame_tokens(self) -> int:
617
+ frame_size = self.get_video_frame_size_with_most_features()
618
+
619
+ return self.get_num_image_tokens(
620
+ frame_size,
621
+ max_slice_nums=self.get_video_max_slice_num(),
622
+ )
623
+
624
+ def get_max_video_tokens(
625
+ self,
626
+ seq_len: int,
627
+ mm_counts: Mapping[str, int],
628
+ ) -> int:
629
+ num_frames = self.get_num_frames_with_most_features(seq_len, mm_counts)
630
+ num_video_tokens_total = self.get_max_video_frame_tokens() * num_frames
631
+ return num_video_tokens_total
632
+
633
+ def get_video_max_slice_num(self) -> int:
634
+ return 1
635
+
636
+ def get_video_frame_size_with_most_features(self) -> ImageSize:
637
+ image_size = getattr(self.get_hf_config(), "image_size", 448)
638
+ max_slice_num = self.get_video_max_slice_num()
639
+ return ImageSize(width=image_size, height=image_size * max_slice_num)
640
+
641
+ def get_max_video_frames(self, max_tokens: int) -> int:
642
+ num_frame_tokens = self.get_max_video_frame_tokens()
643
+ num_frames = max_tokens // num_frame_tokens
644
+ return num_frames
645
+
646
+ def get_num_frames_with_most_features(
647
+ self,
648
+ seq_len: int,
649
+ mm_counts: Mapping[str, int],
650
+ ) -> int:
651
+ max_images = mm_counts.get("image", 0)
652
+ max_videos = mm_counts.get("video", 0)
653
+
654
+ max_image_tokens = self.get_max_image_tokens() * max_images
655
+ max_total_frames = self.get_max_video_frames(seq_len -
656
+ max_image_tokens)
657
+ max_frames_per_video = min(max_total_frames // max(max_videos, 1),
658
+ _MAX_FRAMES_PER_VIDEO)
659
+
660
+ return max(max_frames_per_video, 1)
661
+
662
+
663
+ _I = TypeVar("_I",
664
+ bound=MiniCPMVProcessingInfo,
665
+ default=MiniCPMVProcessingInfo)
666
+
667
+
668
+ class MiniCPMVDummyInputsBuilder(BaseDummyInputsBuilder[_I]):
669
+
670
+ def get_dummy_text(self, mm_counts: Mapping[str, int]) -> str:
671
+ num_images = mm_counts.get("image", 0)
672
+ num_videos = mm_counts.get("video", 0)
673
+
674
+ image_prompt_texts = self.info.image_pattern * num_images
675
+ video_prompt_texts = self.info.video_pattern * num_videos
676
+
677
+ return image_prompt_texts + video_prompt_texts
678
+
679
+ def get_dummy_mm_data(
680
+ self,
681
+ seq_len: int,
682
+ mm_counts: Mapping[str, int],
683
+ ) -> MultiModalDataDict:
684
+ num_images = mm_counts.get("image", 0)
685
+ num_videos = mm_counts.get("video", 0)
686
+
687
+ image_width, image_height = \
688
+ self.info.get_image_size_with_most_features()
689
+ video_width, video_height = \
690
+ self.info.get_video_frame_size_with_most_features()
691
+ num_video_frames = \
692
+ self.info.get_num_frames_with_most_features(seq_len, mm_counts)
693
+
694
+ return {
695
+ "image":
696
+ self._get_dummy_images(width=image_width,
697
+ height=image_height,
698
+ num_images=num_images),
699
+ "video": [
700
+ self._get_dummy_images(width=video_width,
701
+ height=video_height,
702
+ num_images=num_video_frames)
703
+ ] * num_videos,
704
+ }
705
+
706
+
707
+ class MiniCPMVMultiModalProcessor(BaseMultiModalProcessor[_I]):
708
+
709
+ def _get_data_parser(self) -> MultiModalDataParser:
710
+ return MiniCPMVMultiModalDataParser()
711
+
712
+ def get_image_prompt_texts(self,
713
+ image_size: ImageSize,
714
+ image_idx: int = 0) -> str:
715
+ return self.info.get_slice_image_placeholder(
716
+ image_size,
717
+ image_idx=image_idx,
718
+ )
719
+
720
+ def get_video_prompt_texts(self, image_size: ImageSize,
721
+ num_frames: int) -> str:
722
+ return self.info.get_slice_image_placeholder(
723
+ image_size=image_size,
724
+ image_idx=0,
725
+ max_slice_nums=self.info.get_video_max_slice_num(),
726
+ use_image_id=False,
727
+ ) * num_frames
728
+
729
+ def process_images(
730
+ self,
731
+ mm_data: Mapping[str, object],
732
+ mm_kwargs: Mapping[str, object],
733
+ tok_kwargs: Mapping[str, object],
734
+ ) -> Mapping[str, NestedTensors]:
735
+ if (images := mm_data.get("images")) is None:
736
+ return {}
737
+
738
+ parsed_images = (self._get_data_parser().parse_mm_data({
739
+ "image": images
740
+ }).get_items("image",
741
+ (MiniCPMVImageEmbeddingItems, ImageProcessorItems)))
742
+
743
+ if isinstance(parsed_images, MiniCPMVImageEmbeddingItems):
744
+ image_inputs = {}
745
+ else:
746
+ image_inputs = self._base_call_hf_processor(
747
+ prompts=[self.info.image_pattern] * len(parsed_images),
748
+ mm_data={"images": [[image] for image in parsed_images]},
749
+ mm_kwargs=mm_kwargs,
750
+ tok_kwargs=tok_kwargs,
751
+ out_keys={"pixel_values", "image_sizes", "tgt_sizes"},
752
+ )
753
+
754
+ tokenizer = self.info.get_tokenizer()
755
+ unk_token_id = tokenizer.get_vocab()["<unk>"]
756
+ image_inputs["image_token_id"] = torch.tensor(unk_token_id)
757
+
758
+ return image_inputs
759
+
760
+ def process_videos(
761
+ self,
762
+ mm_data: Mapping[str, object],
763
+ mm_kwargs: Mapping[str, object],
764
+ tok_kwargs: Mapping[str, object],
765
+ ) -> Mapping[str, NestedTensors]:
766
+ if (videos := mm_data.get("videos")) is None:
767
+ return {}
768
+
769
+ parsed_videos = (self._get_data_parser().parse_mm_data({
770
+ "video": videos
771
+ }).get_items("video",
772
+ (MiniCPMVVideoEmbeddingItems, VideoProcessorItems)))
773
+
774
+ if isinstance(parsed_videos, MiniCPMVVideoEmbeddingItems):
775
+ video_inputs = {}
776
+ else:
777
+ video_inputs = self._base_call_hf_processor(
778
+ prompts=[
779
+ self.info.image_pattern * len(video)
780
+ for video in parsed_videos
781
+ ],
782
+ mm_data={"images": list(parsed_videos)},
783
+ mm_kwargs={
784
+ **mm_kwargs,
785
+ "max_slice_nums":
786
+ self.info.get_video_max_slice_num(),
787
+ },
788
+ tok_kwargs=tok_kwargs,
789
+ out_keys={"pixel_values", "image_sizes", "tgt_sizes"},
790
+ )
791
+
792
+ video_inputs = {f"video_{k}": v for k, v in video_inputs.items()}
793
+
794
+ tokenizer = self.info.get_tokenizer()
795
+ unk_token_id = tokenizer.get_vocab()["<unk>"]
796
+ video_inputs["video_token_id"] = torch.tensor(unk_token_id)
797
+
798
+ return video_inputs
799
+
800
+ def process_mm_inputs(
801
+ self,
802
+ mm_data: Mapping[str, object],
803
+ mm_kwargs: Mapping[str, object],
804
+ tok_kwargs: Mapping[str, object],
805
+ ) -> Mapping[str, NestedTensors]:
806
+ return {
807
+ **self.process_images(mm_data, mm_kwargs, tok_kwargs),
808
+ **self.process_videos(mm_data, mm_kwargs, tok_kwargs),
809
+ }
810
+
811
+ def _base_call_hf_processor(
812
+ self,
813
+ prompts: list[str],
814
+ mm_data: Mapping[str, Sequence[object]],
815
+ mm_kwargs: Mapping[str, object],
816
+ tok_kwargs: Mapping[str, object],
817
+ *,
818
+ out_keys: set[str],
819
+ ) -> dict[str, NestedTensors]:
820
+ # This processor supports zipping prompt and mm_data together
821
+ if self.info.get_model_version() in {(2, 6), (4, 0), (4, 5)}:
822
+ inputs = super()._call_hf_processor(
823
+ prompt=prompts, # type: ignore
824
+ mm_data=mm_data,
825
+ mm_kwargs=mm_kwargs,
826
+ tok_kwargs=tok_kwargs,
827
+ )
828
+ else:
829
+ inputs = defaultdict[str, list[torch.Tensor]](list)
830
+
831
+ for i, prompt in enumerate(prompts):
832
+ inputs_one = super()._call_hf_processor(
833
+ prompt=prompt,
834
+ mm_data={
835
+ k: v[i]
836
+ for k, v in mm_data.items()
837
+ },
838
+ mm_kwargs=mm_kwargs,
839
+ tok_kwargs=tok_kwargs,
840
+ )
841
+
842
+ for k, v in inputs_one.items():
843
+ assert len(v) == 1, (k, len(v))
844
+ inputs[k].append(v[0])
845
+
846
+ return {k: inputs[k] for k in out_keys}
847
+
848
+ def _call_hf_processor(
849
+ self,
850
+ prompt: str,
851
+ mm_data: Mapping[str, object],
852
+ mm_kwargs: Mapping[str, object],
853
+ tok_kwargs: Mapping[str, object],
854
+ ) -> BatchFeature:
855
+ tokenizer = self.info.get_tokenizer()
856
+
857
+ input_ids = torch.tensor([tokenizer.encode(prompt, **tok_kwargs)])
858
+ mm_inputs = self.process_mm_inputs(mm_data, mm_kwargs, tok_kwargs)
859
+
860
+ return BatchFeature({
861
+ "input_ids": input_ids,
862
+ **mm_inputs,
863
+ })
864
+
865
+ def _hf_processor_applies_updates(
866
+ self,
867
+ prompt_text: str,
868
+ mm_items: MultiModalDataItems,
869
+ hf_processor_mm_kwargs: Mapping[str, object],
870
+ tokenization_kwargs: Mapping[str, object],
871
+ ) -> bool:
872
+ return False
873
+
874
+ def _get_prompt_updates(
875
+ self,
876
+ mm_items: MultiModalDataItems,
877
+ hf_processor_mm_kwargs: Mapping[str, object],
878
+ out_mm_kwargs: MultiModalKwargsItems,
879
+ ) -> Sequence[PromptUpdate]:
880
+ placeholders = [("image", self.info.image_pattern),
881
+ ("video", self.info.video_pattern)]
882
+
883
+ # hard code for inconsistency of encode-decode image_pattern
884
+ additional_placeholders = []
885
+ tokenizer = self.info.get_tokenizer()
886
+ for modality, pattern in placeholders:
887
+ sub_pattern = tokenizer.decode(
888
+ tokenizer.encode(pattern, add_special_tokens=False))
889
+ if sub_pattern != pattern:
890
+ additional_placeholders.append((modality, sub_pattern))
891
+ placeholders += additional_placeholders
892
+
893
+ def get_image_replacement(item_idx: int):
894
+ images = mm_items.get_items(
895
+ "image", (MiniCPMVImageEmbeddingItems, ImageProcessorItems))
896
+
897
+ image_size = images.get_image_size(item_idx)
898
+
899
+ return PromptUpdateDetails.select_text(
900
+ self.get_image_prompt_texts(image_size, item_idx),
901
+ "<unk>",
902
+ )
903
+
904
+ def get_video_replacement(item_idx: int):
905
+ videos = mm_items.get_items(
906
+ "video", (MiniCPMVVideoEmbeddingItems, VideoProcessorItems))
907
+
908
+ frame_size = videos.get_frame_size(item_idx)
909
+ num_frames = videos.get_num_frames(item_idx)
910
+
911
+ return PromptUpdateDetails.select_text(
912
+ self.get_video_prompt_texts(frame_size, num_frames),
913
+ "<unk>",
914
+ )
915
+
916
+ get_replacement = {
917
+ "image": get_image_replacement,
918
+ "video": get_video_replacement,
919
+ }
920
+
921
+ return [
922
+ PromptReplacement(modality=modality,
923
+ target=pattern,
924
+ replacement=get_replacement[modality])
925
+ for modality, pattern in placeholders
926
+ ]
927
+
928
+ def _recompute_cached_prompt_update(
929
+ self,
930
+ cached_update: ResolvedPromptUpdate,
931
+ new_item_idx: int,
932
+ ) -> ResolvedPromptUpdate:
933
+ new_update = super()._recompute_cached_prompt_update(
934
+ cached_update,
935
+ new_item_idx,
936
+ )
937
+
938
+ if cached_update.modality == "image":
939
+ tokenizer = self.info.get_tokenizer()
940
+ image_processor = self.info.get_image_processor()
941
+ version = self.info.get_model_version()
942
+
943
+ text = _seq2text(tokenizer, cached_update.content.full)
944
+ prev_item_idx = cached_update.item_idx
945
+
946
+ if version == (2, 0) or version == (2, 5):
947
+ im_start = image_processor.im_start_token
948
+ im_end = image_processor.im_end_token
949
+ else:
950
+ im_start = image_processor.im_id_start
951
+ im_end = image_processor.im_id_end
952
+
953
+ new_update = new_update.with_content(
954
+ PromptUpdateDetails.select_text(
955
+ text.replace(
956
+ f"{im_start}{prev_item_idx}{im_end}",
957
+ f"{im_start}{new_item_idx}{im_end}",
958
+ 1,
959
+ ),
960
+ "<unk>",
961
+ ))
962
+
963
+ return new_update
964
+
965
+ def _get_mm_fields_config(
966
+ self,
967
+ hf_inputs: BatchFeature,
968
+ hf_processor_mm_kwargs: Mapping[str, object],
969
+ ) -> Mapping[str, MultiModalFieldConfig]:
970
+ return _minicpmv_field_config(hf_inputs)
971
+
972
+
973
+ class MiniCPMVBaseModel(nn.Module, SupportsMultiModal, SupportsPP):
974
+ """
975
+ The abstract class of MiniCPMV can only be inherited, but cannot be
976
+ instantiated.
977
+ """
978
+
979
+ supports_encoder_tp_data = True
980
+
981
+ @classmethod
982
+ def get_placeholder_str(cls, modality: str, i: int) -> Optional[str]:
983
+ if modality.startswith("image"):
984
+ return "(<image>./</image>)"
985
+ if modality.startswith("video"):
986
+ return "(<video>./</video>)"
987
+
988
+ raise ValueError("Only image or video modality is supported")
989
+
990
+ def __init__(self, *, vllm_config: VllmConfig, prefix: str = ""):
991
+ config = vllm_config.model_config.hf_config
992
+ multimodal_config = vllm_config.model_config.multimodal_config
993
+ quant_config = vllm_config.quant_config
994
+ self.use_data_parallel = multimodal_config.mm_encoder_tp_mode == "data"
995
+ super().__init__()
996
+ # All MiniCPM-V models disable `tie_word_embeddings` but
997
+ # `PretrainedConfig.tie_word_embeddings` defaults to True; we cannot
998
+ # check `tie_word_embeddings` until vLLM integrate MiniCPM-V model
999
+ # and config class
1000
+ self.config = config
1001
+ self.multimodal_config = multimodal_config
1002
+
1003
+ self.version = get_version_by_config(self.config)
1004
+ self.llm = self.init_llm(vllm_config=vllm_config,
1005
+ prefix=maybe_prefix(prefix, "llm"))
1006
+ self.vpm = self.init_vision_module(config,
1007
+ quant_config,
1008
+ prefix=maybe_prefix(prefix, "vpm"))
1009
+ self.vision_dim = (self.vpm.embed_dim if self.version == (2, 0) else
1010
+ self.vpm.embeddings.embed_dim)
1011
+ self.embed_dim = self.config.hidden_size
1012
+
1013
+ self.resampler = self.init_resampler(self.embed_dim,
1014
+ self.vision_dim,
1015
+ quant_config=quant_config,
1016
+ prefix=maybe_prefix(
1017
+ prefix, "resampler"))
1018
+
1019
+ self.mm_token_ids = set[int]()
1020
+ self.make_empty_intermediate_tensors = (
1021
+ self.llm.make_empty_intermediate_tensors)
1022
+
1023
+ def _parse_and_validate_vision_input(
1024
+ self,
1025
+ modality: str,
1026
+ **kwargs: object,
1027
+ ) -> Optional[MiniCPMVImageInputs]:
1028
+ pixel_values = kwargs.pop("pixel_values", None)
1029
+ image_embeds = kwargs.pop("image_embeds", None)
1030
+
1031
+ if pixel_values is None and image_embeds is None:
1032
+ return None
1033
+
1034
+ image_token_id = kwargs.pop("image_token_id")
1035
+ if image_token_id is not None:
1036
+ assert isinstance(image_token_id, torch.Tensor)
1037
+ self.mm_token_ids.add(image_token_id.flatten().unique().item())
1038
+
1039
+ if image_embeds is not None:
1040
+ if not isinstance(image_embeds, (torch.Tensor, list)):
1041
+ raise ValueError(
1042
+ f"Incorrect type of image_embeds for {modality=}. "
1043
+ f"Got type: {type(image_embeds)}")
1044
+
1045
+ image_embeds_flat = flatten_bn(image_embeds)
1046
+
1047
+ return MiniCPMVImageEmbeddingInputs(
1048
+ type="image_embeds",
1049
+ image_embeds=image_embeds_flat,
1050
+ )
1051
+
1052
+ if not isinstance(pixel_values, (torch.Tensor, list)):
1053
+ raise ValueError(
1054
+ f"Incorrect type of pixel_values for {modality=}. "
1055
+ f"Got type: {type(pixel_values)}")
1056
+
1057
+ tgt_sizes = kwargs.pop("tgt_sizes")
1058
+ if not isinstance(tgt_sizes, (torch.Tensor, list)):
1059
+ raise ValueError(f"Incorrect type of tgt_sizes for {modality=}. "
1060
+ f"Got type: {type(tgt_sizes)}")
1061
+
1062
+ num_slices = [[len(p) for p in ps] for ps in pixel_values]
1063
+ num_slices_flat = flatten_bn(torch.tensor(num_slices))
1064
+
1065
+ pixel_values_flat = flatten_bn(flatten_2d_lists(pixel_values))
1066
+ tgt_sizes_flat = flatten_bn(flatten_2d_lists(tgt_sizes), concat=True)
1067
+
1068
+ return MiniCPMVImagePixelInputs(
1069
+ type="pixel_values",
1070
+ pixel_values=pixel_values_flat,
1071
+ tgt_sizes=tgt_sizes_flat,
1072
+ num_slices=num_slices_flat,
1073
+ )
1074
+
1075
+ def _parse_and_validate_multimodal_inputs(self, **kwargs: object) -> dict:
1076
+ modalities = {}
1077
+
1078
+ # Preserve the order of modalities if there are multiple of them
1079
+ # from the order of kwargs.
1080
+ for input_key in kwargs:
1081
+ if input_key in ("pixel_values",
1082
+ "image_embeds") and "images" not in modalities:
1083
+ modalities["images"] = self._parse_and_validate_vision_input(
1084
+ "images", **kwargs)
1085
+ if input_key in ("video_pixel_values",
1086
+ "video_embeds") and "videos" not in modalities:
1087
+
1088
+ def _image_key(video_key: str):
1089
+ if video_key == "video_token_id":
1090
+ return "image_token_id"
1091
+
1092
+ return video_key.removeprefix("video_")
1093
+
1094
+ modalities["videos"] = self._parse_and_validate_vision_input(
1095
+ "videos", **{
1096
+ _image_key(k): v
1097
+ for k, v in kwargs.items()
1098
+ })
1099
+
1100
+ return modalities
1101
+
1102
+ def _process_vision_input(
1103
+ self,
1104
+ image_input: MiniCPMVImageInputs,
1105
+ ) -> Union[torch.Tensor, list[torch.Tensor], tuple[torch.Tensor, ...]]:
1106
+ if image_input["type"] == "image_embeds":
1107
+ return image_input["image_embeds"]
1108
+
1109
+ image_features_flat = self.get_vision_hidden_states(image_input)
1110
+
1111
+ num_slices = image_input["num_slices"]
1112
+ return [
1113
+ e.flatten(0, 1)
1114
+ for e in image_features_flat.split(num_slices.tolist())
1115
+ ]
1116
+
1117
+ def _process_multimodal_inputs(self, modalities: dict):
1118
+ # The result multimodal_embeddings is tuple of tensors, with each
1119
+ # tensor corresponding to a multimodal data item (image or video).
1120
+ multimodal_embeddings: tuple[torch.Tensor, ...] = ()
1121
+
1122
+ # NOTE: It is important to iterate over the keys in this dictionary
1123
+ # to preserve the order of the modalities.
1124
+ for modality in modalities:
1125
+ if modality == "images":
1126
+ image_input = modalities["images"]
1127
+ image_features = self._process_vision_input(image_input)
1128
+ multimodal_embeddings += tuple(image_features)
1129
+ if modality == "videos":
1130
+ video_input = modalities["videos"]
1131
+ video_features = self._process_vision_input(video_input)
1132
+ multimodal_embeddings += tuple(video_features)
1133
+
1134
+ return multimodal_embeddings
1135
+
1136
+ def get_language_model(self) -> torch.nn.Module:
1137
+ return self.llm
1138
+
1139
+ def get_multimodal_embeddings(self,
1140
+ **kwargs: object) -> MultiModalEmbeddings:
1141
+ modalities = self._parse_and_validate_multimodal_inputs(**kwargs)
1142
+ if not modalities:
1143
+ return []
1144
+
1145
+ return self._process_multimodal_inputs(modalities)
1146
+
1147
+ def get_input_embeddings(
1148
+ self,
1149
+ input_ids: torch.Tensor,
1150
+ multimodal_embeddings: Optional[MultiModalEmbeddings] = None,
1151
+ ) -> torch.Tensor:
1152
+ inputs_embeds = self.llm.get_input_embeddings(input_ids)
1153
+ if multimodal_embeddings is not None \
1154
+ and len(multimodal_embeddings) != 0:
1155
+ assert len(self.mm_token_ids) > 0
1156
+ inputs_embeds = merge_multimodal_embeddings(
1157
+ input_ids,
1158
+ inputs_embeds,
1159
+ multimodal_embeddings,
1160
+ list(self.mm_token_ids),
1161
+ )
1162
+ return inputs_embeds
1163
+
1164
+ def forward(
1165
+ self,
1166
+ input_ids: torch.Tensor,
1167
+ positions: torch.Tensor,
1168
+ intermediate_tensors: Optional[IntermediateTensors] = None,
1169
+ inputs_embeds: Optional[torch.Tensor] = None,
1170
+ **kwargs: Any,
1171
+ ) -> torch.Tensor:
1172
+ if intermediate_tensors is not None:
1173
+ inputs_embeds = None
1174
+
1175
+ # NOTE: In v1, inputs_embeds is always generated at model runner from
1176
+ # `get_multimodal_embeddings` and `get_input_embeddings`, this
1177
+ # condition is only for v0 compatibility.
1178
+ elif inputs_embeds is None:
1179
+ vision_embeddings = self.get_multimodal_embeddings(**kwargs)
1180
+
1181
+ inputs_embeds = self.get_input_embeddings(input_ids,
1182
+ vision_embeddings)
1183
+ input_ids = None
1184
+
1185
+ hidden_states = self.llm.model(
1186
+ input_ids=input_ids,
1187
+ positions=positions,
1188
+ intermediate_tensors=intermediate_tensors,
1189
+ inputs_embeds=inputs_embeds,
1190
+ )
1191
+ return hidden_states
1192
+
1193
+ def compute_logits(
1194
+ self,
1195
+ hidden_states: torch.Tensor,
1196
+ ) -> Optional[torch.Tensor]:
1197
+ return self.llm.compute_logits(hidden_states)
1198
+
1199
+ def load_weights(self, weights: Iterable[tuple[str,
1200
+ torch.Tensor]]) -> set[str]:
1201
+ loader = AutoWeightsLoader(self)
1202
+ return loader.load_weights(weights)
1203
+
1204
+ def get_mm_mapping(self) -> MultiModelKeys:
1205
+ """
1206
+ Get the module prefix in multimodal models
1207
+ """
1208
+ return MultiModelKeys.from_string_field(language_model="llm",
1209
+ connector="resampler",
1210
+ tower_model="vpm")
1211
+
1212
+ def init_llm(
1213
+ self,
1214
+ vllm_config: VllmConfig,
1215
+ prefix: str = "",
1216
+ ) -> nn.Module:
1217
+ raise NotImplementedError
1218
+
1219
+ def init_vision_module(
1220
+ self,
1221
+ config: PretrainedConfig,
1222
+ quant_config: Optional[QuantizationConfig],
1223
+ prefix: str = "",
1224
+ ) -> nn.Module:
1225
+ raise NotImplementedError
1226
+
1227
+ def init_resampler(self,
1228
+ embed_dim: int,
1229
+ vision_dim: int,
1230
+ quant_config: Optional[QuantizationConfig] = None,
1231
+ prefix: str = "") -> nn.Module:
1232
+ raise NotImplementedError
1233
+
1234
+ def get_vision_hidden_states(
1235
+ self, data: MiniCPMVImagePixelInputs) -> torch.Tensor:
1236
+ raise NotImplementedError
1237
+
1238
+
1239
+ class MiniCPMV2_0(MiniCPMVBaseModel):
1240
+
1241
+ supports_encoder_tp_data = False
1242
+
1243
+ def __init__(self, *, vllm_config: VllmConfig, prefix: str = ""):
1244
+ super().__init__(vllm_config=vllm_config, prefix=prefix)
1245
+ assert self.version == (2, 0)
1246
+
1247
+ def init_llm(
1248
+ self,
1249
+ vllm_config: VllmConfig,
1250
+ prefix: str = "",
1251
+ ) -> nn.Module:
1252
+ return MiniCPMForCausalLM(vllm_config=vllm_config, prefix=prefix)
1253
+
1254
+ def init_vision_module(
1255
+ self,
1256
+ config: PretrainedConfig,
1257
+ quant_config: Optional[QuantizationConfig],
1258
+ prefix: str = "",
1259
+ ) -> nn.Module:
1260
+ # TODO: refactor vision model through timm wrapper from transformers
1261
+ try:
1262
+ import timm
1263
+ except ImportError:
1264
+ raise ImportError("Please install timm==0.9.10") from ImportError
1265
+
1266
+ with set_default_torch_dtype(torch.float16):
1267
+ model = timm.create_model(
1268
+ "vit_so400m_patch14_siglip_384.webli",
1269
+ pretrained=False,
1270
+ num_classes=0,
1271
+ dynamic_img_size=True,
1272
+ dynamic_img_pad=True,
1273
+ )
1274
+
1275
+ model = model.to(dtype=torch.get_default_dtype())
1276
+
1277
+ if (isinstance(model, timm.models.VisionTransformer)
1278
+ and model.attn_pool is not None):
1279
+ model.attn_pool = torch.nn.Identity()
1280
+
1281
+ if self.config.drop_vision_last_layer:
1282
+ model.blocks = model.blocks[:-1]
1283
+
1284
+ return model
1285
+
1286
+ def init_resampler(self,
1287
+ embed_dim: int,
1288
+ vision_dim: int,
1289
+ quant_config: Optional[QuantizationConfig] = None,
1290
+ prefix: str = "") -> nn.Module:
1291
+ with set_default_torch_dtype(torch.float16):
1292
+ resampler = Resampler2(embed_dim=embed_dim,
1293
+ num_heads=embed_dim // 128,
1294
+ grid_size=int(
1295
+ math.sqrt(self.config.query_num)),
1296
+ kv_dim=vision_dim,
1297
+ adaptive=False,
1298
+ do_post_projection=True,
1299
+ quant_config=quant_config,
1300
+ prefix=prefix)
1301
+
1302
+ return resampler.to(device=current_platform.device_type,
1303
+ dtype=torch.get_default_dtype())
1304
+
1305
+ def get_vision_hidden_states(
1306
+ self, data: MiniCPMVImagePixelInputs) -> torch.Tensor:
1307
+ pixel_values = data["pixel_values"]
1308
+
1309
+ P_h, P_w = self.vpm.patch_embed.patch_size
1310
+ dtype: torch.dtype = self.vpm.pos_embed.data.dtype
1311
+ num_prefix_tokens = getattr(self.vpm, "num_prefix_tokens", 0)
1312
+
1313
+ res = list[torch.Tensor]()
1314
+ for pixel_value in pixel_values:
1315
+ H, W = pixel_value[0].shape[-2:]
1316
+ tgt_size = (math.ceil(H / P_h), math.ceil(W / P_w))
1317
+ vision_embedding = self.vpm.forward_features(
1318
+ pixel_value.unsqueeze(0).type(dtype))
1319
+
1320
+ if num_prefix_tokens > 0:
1321
+ vision_embedding = vision_embedding[:, num_prefix_tokens:]
1322
+ res.append(self.resampler(vision_embedding, tgt_size))
1323
+
1324
+ return torch.vstack(res)
1325
+
1326
+
1327
+ class MiniCPMV2_5(MiniCPMVBaseModel, SupportsLoRA):
1328
+ packed_modules_mapping = {
1329
+ "qkv_proj": [
1330
+ "q_proj",
1331
+ "k_proj",
1332
+ "v_proj",
1333
+ ],
1334
+ "gate_up_proj": [
1335
+ "gate_proj",
1336
+ "up_proj",
1337
+ ],
1338
+ }
1339
+
1340
+ def __init__(self, *, vllm_config: VllmConfig, prefix: str = ""):
1341
+ super().__init__(vllm_config=vllm_config, prefix=prefix)
1342
+ assert self.version == (2, 5)
1343
+
1344
+ def init_llm(
1345
+ self,
1346
+ vllm_config: VllmConfig,
1347
+ prefix: str = "",
1348
+ ) -> nn.Module:
1349
+ return LlamaForCausalLM(vllm_config=vllm_config, prefix=prefix)
1350
+
1351
+ def init_vision_module(
1352
+ self,
1353
+ config: PretrainedConfig,
1354
+ quant_config: Optional[QuantizationConfig],
1355
+ prefix: str = "",
1356
+ ) -> nn.Module:
1357
+ model = Idefics2VisionTransformer(
1358
+ config.vision_config,
1359
+ quant_config=quant_config,
1360
+ prefix=prefix,
1361
+ use_data_parallel=self.use_data_parallel,
1362
+ )
1363
+ if self.config.drop_vision_last_layer:
1364
+ model.encoder.layers = model.encoder.layers[:-1]
1365
+ return model
1366
+
1367
+ def init_resampler(self,
1368
+ embed_dim: int,
1369
+ vision_dim: int,
1370
+ quant_config: Optional[QuantizationConfig] = None,
1371
+ prefix: str = "") -> nn.Module:
1372
+ with set_default_torch_dtype(torch.float16):
1373
+ resampler = Resampler2_5(num_queries=self.config.query_num,
1374
+ embed_dim=embed_dim,
1375
+ num_heads=embed_dim // 128,
1376
+ kv_dim=vision_dim,
1377
+ quant_config=quant_config,
1378
+ prefix=prefix)
1379
+
1380
+ return resampler.to(device=current_platform.device_type,
1381
+ dtype=torch.get_default_dtype())
1382
+
1383
+ def get_vision_hidden_states(
1384
+ self, data: MiniCPMVImagePixelInputs) -> torch.Tensor:
1385
+ pixel_values = data["pixel_values"]
1386
+ tgt_sizes = data["tgt_sizes"]
1387
+
1388
+ B = len(pixel_values)
1389
+ P = pixel_values[0].shape[-2]
1390
+ L = max(item.shape[-1] for item in pixel_values)
1391
+ device = pixel_values[0].device
1392
+ dtype = pixel_values[0].dtype
1393
+
1394
+ all_pixel_values = torch.zeros((B, 3, P, L),
1395
+ dtype=dtype,
1396
+ device=device)
1397
+ for i, pixel_values_item in enumerate(pixel_values):
1398
+ L_item = pixel_values_item.shape[-1]
1399
+ all_pixel_values[i, ..., :L_item] = pixel_values_item
1400
+
1401
+ num_patches = tgt_sizes.prod(-1)
1402
+ max_patches = num_patches.max().item()
1403
+ assert isinstance(max_patches, int)
1404
+
1405
+ patch_attn_mask = torch.zeros((B, max_patches),
1406
+ dtype=torch.bool,
1407
+ device=device)
1408
+ for i, num_patches_item in enumerate(num_patches):
1409
+ patch_attn_mask[i, :num_patches_item] = True
1410
+
1411
+ vision_embedding = self.vpm(
1412
+ all_pixel_values,
1413
+ patch_attention_mask=patch_attn_mask.unsqueeze(1),
1414
+ tgt_sizes=None,
1415
+ )
1416
+
1417
+ return self.resampler(vision_embedding, tgt_sizes)
1418
+
1419
+
1420
+ class MiniCPMV2_6(MiniCPMVBaseModel, SupportsLoRA):
1421
+ packed_modules_mapping = {
1422
+ "qkv_proj": [
1423
+ "q_proj",
1424
+ "k_proj",
1425
+ "v_proj",
1426
+ ],
1427
+ "gate_up_proj": [
1428
+ "gate_proj",
1429
+ "up_proj",
1430
+ ],
1431
+ }
1432
+
1433
+ def __init__(self, *, vllm_config: VllmConfig, prefix: str = ""):
1434
+ super().__init__(vllm_config=vllm_config, prefix=prefix)
1435
+ assert self.version == (2, 6)
1436
+
1437
+ def init_llm(
1438
+ self,
1439
+ vllm_config: VllmConfig,
1440
+ prefix: str = "",
1441
+ ) -> nn.Module:
1442
+ return Qwen2ForCausalLM(vllm_config=vllm_config, prefix=prefix)
1443
+
1444
+ def init_vision_module(
1445
+ self,
1446
+ config: PretrainedConfig,
1447
+ quant_config: Optional[QuantizationConfig] = None,
1448
+ prefix: str = "",
1449
+ ) -> nn.Module:
1450
+ model = Idefics2VisionTransformer(
1451
+ config.vision_config,
1452
+ quant_config=quant_config,
1453
+ prefix=prefix,
1454
+ use_data_parallel=self.use_data_parallel,
1455
+ )
1456
+ if self.config.drop_vision_last_layer:
1457
+ model.encoder.layers = model.encoder.layers[:-1]
1458
+ return model
1459
+
1460
+ def init_resampler(self,
1461
+ embed_dim: int,
1462
+ vision_dim: int,
1463
+ quant_config: Optional[QuantizationConfig] = None,
1464
+ prefix: str = "") -> nn.Module:
1465
+ with set_default_torch_dtype(torch.float16):
1466
+ # The resampler in 2.6 remains consistent with the one in 2.5.
1467
+ resampler = Resampler2_5(num_queries=self.config.query_num,
1468
+ embed_dim=embed_dim,
1469
+ num_heads=embed_dim // 128,
1470
+ kv_dim=vision_dim,
1471
+ quant_config=quant_config,
1472
+ prefix=prefix)
1473
+
1474
+ return resampler.to(device=current_platform.device_type,
1475
+ dtype=torch.get_default_dtype())
1476
+
1477
+ def get_vision_hidden_states(
1478
+ self, data: MiniCPMVImagePixelInputs) -> torch.Tensor:
1479
+ pixel_values = data["pixel_values"]
1480
+ tgt_sizes = data["tgt_sizes"]
1481
+
1482
+ B = len(pixel_values)
1483
+ P = pixel_values[0].shape[-2]
1484
+ L = max(item.shape[-1] for item in pixel_values)
1485
+ device = pixel_values[0].device
1486
+ dtype = pixel_values[0].dtype
1487
+
1488
+ all_pixel_values = torch.zeros((B, 3, P, L),
1489
+ dtype=dtype,
1490
+ device=device)
1491
+ for i, pixel_values_item in enumerate(pixel_values):
1492
+ L_item = pixel_values_item.shape[-1]
1493
+ all_pixel_values[i, ..., :L_item] = pixel_values_item
1494
+
1495
+ num_patches = tgt_sizes.prod(-1)
1496
+ max_patches = num_patches.max().item()
1497
+ assert isinstance(max_patches, int)
1498
+
1499
+ patch_attn_mask = torch.zeros((B, max_patches),
1500
+ dtype=torch.bool,
1501
+ device=device)
1502
+ for i, num_patches_item in enumerate(num_patches):
1503
+ patch_attn_mask[i, :num_patches_item] = True
1504
+
1505
+ vision_embedding = self.vpm(
1506
+ all_pixel_values,
1507
+ patch_attention_mask=patch_attn_mask.unsqueeze(1),
1508
+ tgt_sizes=tgt_sizes,
1509
+ )
1510
+
1511
+ return self.resampler(vision_embedding, tgt_sizes)
1512
+
1513
+ def load_weights(self, weights: Iterable[tuple[str,
1514
+ torch.Tensor]]) -> set[str]:
1515
+ loader = AutoWeightsLoader(self,
1516
+ skip_prefixes=["apm.", "audio", "tts"])
1517
+ return loader.load_weights(weights)
1518
+
1519
+
1520
+ class MiniCPMV4_0(MiniCPMVBaseModel, SupportsLoRA):
1521
+ packed_modules_mapping = {
1522
+ "qkv_proj": [
1523
+ "q_proj",
1524
+ "k_proj",
1525
+ "v_proj",
1526
+ ],
1527
+ "gate_up_proj": [
1528
+ "gate_proj",
1529
+ "up_proj",
1530
+ ],
1531
+ }
1532
+
1533
+ def __init__(self, *, vllm_config: VllmConfig, prefix: str = ""):
1534
+ super().__init__(vllm_config=vllm_config, prefix=prefix)
1535
+ assert self.version == (4, 0)
1536
+
1537
+ def _maybe_ignore_quant_config(self, quant_config: QuantizationConfig):
1538
+ if isinstance(quant_config, (AWQConfig, AWQMarlinConfig)):
1539
+ return None
1540
+ return quant_config
1541
+
1542
+ def init_llm(
1543
+ self,
1544
+ vllm_config: VllmConfig,
1545
+ prefix: str = "",
1546
+ ) -> nn.Module:
1547
+ return LlamaForCausalLM(vllm_config=vllm_config, prefix=prefix)
1548
+
1549
+ def init_vision_module(
1550
+ self,
1551
+ config: PretrainedConfig,
1552
+ quant_config: Optional[QuantizationConfig] = None,
1553
+ prefix: str = "",
1554
+ ) -> nn.Module:
1555
+ quant_config = self._maybe_ignore_quant_config(quant_config)
1556
+ model = Idefics2VisionTransformer(
1557
+ config.vision_config,
1558
+ quant_config=quant_config,
1559
+ prefix=prefix,
1560
+ use_data_parallel=self.use_data_parallel,
1561
+ )
1562
+ if self.config.drop_vision_last_layer:
1563
+ model.encoder.layers = model.encoder.layers[:-1]
1564
+ return model
1565
+
1566
+ def init_resampler(
1567
+ self,
1568
+ embed_dim: int,
1569
+ vision_dim: int,
1570
+ quant_config: Optional[QuantizationConfig] = None,
1571
+ prefix: str = "",
1572
+ ) -> nn.Module:
1573
+ quant_config = self._maybe_ignore_quant_config(quant_config)
1574
+ with set_default_torch_dtype(torch.float16):
1575
+ # The resampler in 4.0 remains consistent with the one in 2.5/2.6.
1576
+ resampler = Resampler2_5(num_queries=self.config.query_num,
1577
+ embed_dim=embed_dim,
1578
+ num_heads=embed_dim // 128,
1579
+ kv_dim=vision_dim,
1580
+ quant_config=quant_config,
1581
+ prefix=prefix)
1582
+
1583
+ return resampler.to(device=current_platform.device_type,
1584
+ dtype=torch.get_default_dtype())
1585
+
1586
+ def get_vision_hidden_states(
1587
+ self, data: MiniCPMVImagePixelInputs) -> torch.Tensor:
1588
+ pixel_values = data["pixel_values"]
1589
+ tgt_sizes = data["tgt_sizes"]
1590
+
1591
+ B = len(pixel_values)
1592
+ P = pixel_values[0].shape[-2]
1593
+ L = max(item.shape[-1] for item in pixel_values)
1594
+ device = pixel_values[0].device
1595
+ dtype = pixel_values[0].dtype
1596
+
1597
+ all_pixel_values = torch.zeros((B, 3, P, L),
1598
+ dtype=dtype,
1599
+ device=device)
1600
+ for i, pixel_values_item in enumerate(pixel_values):
1601
+ L_item = pixel_values_item.shape[-1]
1602
+ all_pixel_values[i, ..., :L_item] = pixel_values_item
1603
+
1604
+ num_patches = tgt_sizes.prod(-1)
1605
+ max_patches = num_patches.max().item()
1606
+ assert isinstance(max_patches, int)
1607
+
1608
+ patch_attn_mask = torch.zeros((B, max_patches),
1609
+ dtype=torch.bool,
1610
+ device=device)
1611
+ for i, num_patches_item in enumerate(num_patches):
1612
+ patch_attn_mask[i, :num_patches_item] = True
1613
+
1614
+ vision_embedding = self.vpm(
1615
+ all_pixel_values,
1616
+ patch_attention_mask=patch_attn_mask.unsqueeze(1),
1617
+ tgt_sizes=tgt_sizes,
1618
+ )
1619
+
1620
+ return self.resampler(vision_embedding, tgt_sizes)
1621
+
1622
+ def load_weights(self, weights: Iterable[tuple[str,
1623
+ torch.Tensor]]) -> set[str]:
1624
+ loader = AutoWeightsLoader(self,
1625
+ skip_prefixes=["apm.", "audio", "tts"])
1626
+ return loader.load_weights(weights)
1627
+
1628
+
1629
+ class MiniCPMV4_5(MiniCPMVBaseModel, SupportsLoRA):
1630
+ packed_modules_mapping = {
1631
+ "qkv_proj": [
1632
+ "q_proj",
1633
+ "k_proj",
1634
+ "v_proj",
1635
+ ],
1636
+ "gate_up_proj": [
1637
+ "gate_proj",
1638
+ "up_proj",
1639
+ ],
1640
+ }
1641
+
1642
+ def __init__(self, *, vllm_config: VllmConfig, prefix: str = ""):
1643
+ super().__init__(vllm_config=vllm_config, prefix=prefix)
1644
+ assert self.version == (4, 5)
1645
+
1646
+ def _maybe_ignore_quant_config(self, quant_config: QuantizationConfig):
1647
+ if isinstance(quant_config, (AWQConfig, AWQMarlinConfig)):
1648
+ return None
1649
+ return quant_config
1650
+
1651
+ def init_llm(
1652
+ self,
1653
+ vllm_config: VllmConfig,
1654
+ prefix: str = "",
1655
+ ) -> nn.Module:
1656
+ return Qwen3ForCausalLM(vllm_config=vllm_config, prefix=prefix)
1657
+
1658
+ def init_vision_module(
1659
+ self,
1660
+ config: PretrainedConfig,
1661
+ quant_config: Optional[QuantizationConfig] = None,
1662
+ prefix: str = "",
1663
+ ) -> nn.Module:
1664
+ quant_config = self._maybe_ignore_quant_config(quant_config)
1665
+ model = Idefics2VisionTransformer(
1666
+ config.vision_config,
1667
+ quant_config=quant_config,
1668
+ prefix=prefix,
1669
+ use_data_parallel=self.use_data_parallel,
1670
+ )
1671
+ if self.config.drop_vision_last_layer:
1672
+ model.encoder.layers = model.encoder.layers[:-1]
1673
+ return model
1674
+
1675
+ def init_resampler(
1676
+ self,
1677
+ embed_dim: int,
1678
+ vision_dim: int,
1679
+ quant_config: Optional[QuantizationConfig] = None,
1680
+ prefix: str = "",
1681
+ ) -> nn.Module:
1682
+ quant_config = self._maybe_ignore_quant_config(quant_config)
1683
+ with set_default_torch_dtype(torch.float16):
1684
+ # The resampler in 4.0 remains consistent with the one in 2.5/2.6.
1685
+ resampler = Resampler4_5(num_queries=self.config.query_num,
1686
+ embed_dim=embed_dim,
1687
+ num_heads=embed_dim // 128,
1688
+ kv_dim=vision_dim,
1689
+ quant_config=quant_config,
1690
+ prefix=prefix)
1691
+
1692
+ return resampler.to(device=current_platform.device_type,
1693
+ dtype=torch.get_default_dtype())
1694
+
1695
+ def get_vision_hidden_states(
1696
+ self, data: MiniCPMVImagePixelInputs) -> torch.Tensor:
1697
+ pixel_values = data["pixel_values"]
1698
+ tgt_sizes = data["tgt_sizes"]
1699
+ temporal_ids = data.get('temporal_ids', None)
1700
+
1701
+ B = len(pixel_values)
1702
+ P = pixel_values[0].shape[-2]
1703
+ L = max(item.shape[-1] for item in pixel_values)
1704
+ device = pixel_values[0].device
1705
+ dtype = pixel_values[0].dtype
1706
+
1707
+ all_pixel_values = torch.zeros((B, 3, P, L),
1708
+ dtype=dtype,
1709
+ device=device)
1710
+ all_temporal_ids = None if temporal_ids is None else flatten_2d_lists(
1711
+ temporal_ids)
1712
+ for i, pixel_values_item in enumerate(pixel_values):
1713
+ L_item = pixel_values_item.shape[-1]
1714
+ all_pixel_values[i, ..., :L_item] = pixel_values_item
1715
+
1716
+ num_patches = tgt_sizes.prod(-1)
1717
+ max_patches = num_patches.max().item()
1718
+ assert isinstance(max_patches, int)
1719
+
1720
+ patch_attn_mask = torch.zeros((B, max_patches),
1721
+ dtype=torch.bool,
1722
+ device=device)
1723
+ for i, num_patches_item in enumerate(num_patches):
1724
+ patch_attn_mask[i, :num_patches_item] = True
1725
+
1726
+ vision_embedding = self.vpm(
1727
+ all_pixel_values,
1728
+ patch_attention_mask=patch_attn_mask.unsqueeze(1),
1729
+ tgt_sizes=tgt_sizes,
1730
+ )
1731
+
1732
+ return self.resampler(vision_embedding, tgt_sizes, all_temporal_ids)
1733
+
1734
+ def load_weights(self, weights: Iterable[tuple[str,
1735
+ torch.Tensor]]) -> set[str]:
1736
+ loader = AutoWeightsLoader(self,
1737
+ skip_prefixes=["apm.", "audio", "tts"])
1738
+ return loader.load_weights(weights)
1739
+
1740
+
1741
+ _SUPPORT_VERSION = {
1742
+ (2, 0): MiniCPMV2_0,
1743
+ (2, 5): MiniCPMV2_5,
1744
+ (2, 6): MiniCPMV2_6,
1745
+ (4, 0): MiniCPMV4_0,
1746
+ (4, 5): MiniCPMV4_5,
1747
+ }
1748
+
1749
+
1750
+ @MULTIMODAL_REGISTRY.register_processor(
1751
+ MiniCPMVMultiModalProcessor,
1752
+ info=MiniCPMVProcessingInfo,
1753
+ dummy_inputs=MiniCPMVDummyInputsBuilder)
1754
+ class MiniCPMV(MiniCPMVBaseModel, SupportsMultiModal, SupportsLoRA):
1755
+ """
1756
+ Different versions of MiniCPMV use different visual encoders and LLMs,
1757
+ which is not conducive to the current integration logic of LoRA and
1758
+ bitsandbytes in vLLM. Therefore, it is necessary to separate them.
1759
+ """
1760
+
1761
+ def __new__(cls, *, vllm_config: VllmConfig, prefix: str = ""):
1762
+ config = vllm_config.model_config.hf_config
1763
+ if not hasattr(config, "version"):
1764
+ if config.hidden_size == 2304 and config.query_num == 64:
1765
+ version = (2, 0)
1766
+ else:
1767
+ version = (2, 5)
1768
+ else:
1769
+ version = str(config.version).split(".")
1770
+ version = tuple([int(x) for x in version])
1771
+ # Dispatch class based on version
1772
+ instance_cls = _SUPPORT_VERSION.get(version)
1773
+ if instance_cls is None:
1774
+ supported_versions = ", ".join(
1775
+ [f"{v[0]}.{v[1]}" for v in sorted(_SUPPORT_VERSION.keys())])
1776
+ raise ValueError(f"Currently, MiniCPMV only supports versions "
1777
+ f"{supported_versions}. Got version: {version}")
1778
+
1779
+ # quant_config references base class members,
1780
+ # so update values before init is called
1781
+ cls.packed_modules_mapping.update(instance_cls.packed_modules_mapping)
1782
+ cls.embedding_modules.update(instance_cls.embedding_modules)
1783
+ cls.embedding_padding_modules += instance_cls.embedding_padding_modules
1784
+ return instance_cls(vllm_config=vllm_config, prefix=prefix)