[Bugfix] embed_is_patch for Idefics3 (#15696)

Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>
2025-03-28 23:27:52 +08:00
parent 3b00ff9138
commit 541d1df486
5 changed files with 320 additions and 188 deletions
--- a/vllm/model_executor/models/commandr.py
+++ b/vllm/model_executor/models/commandr.py
@@ -24,7 +24,6 @@
 from typing import Iterable, Optional, Set, Tuple, Union

 import torch
-import torch.utils.checkpoint
 from torch import nn
 from transformers import CohereConfig