mirror of
https://github.com/huggingface/lerobot.git
synced 2026-07-24 10:16:09 +00:00
chore(dependencies): bump wall x to transformers v5
This commit is contained in:
@@ -331,7 +331,7 @@ class Qwen2_5_VLMoEForAction(Qwen2_5_VLForConditionalGeneration):
|
|||||||
force_download=kwargs.get("force_download", False),
|
force_download=kwargs.get("force_download", False),
|
||||||
resume_download=kwargs.get("resume_download"),
|
resume_download=kwargs.get("resume_download"),
|
||||||
proxies=kwargs.get("proxies"),
|
proxies=kwargs.get("proxies"),
|
||||||
use_auth_token=kwargs.get("use_auth_token"),
|
token=kwargs.get("token"),
|
||||||
revision=kwargs.get("revision"),
|
revision=kwargs.get("revision"),
|
||||||
local_files_only=kwargs.get("local_files_only", False),
|
local_files_only=kwargs.get("local_files_only", False),
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -11,7 +11,6 @@ from transformers.activations import ACT2FN
|
|||||||
from transformers.cache_utils import (
|
from transformers.cache_utils import (
|
||||||
Cache,
|
Cache,
|
||||||
DynamicCache,
|
DynamicCache,
|
||||||
SlidingWindowCache,
|
|
||||||
StaticCache,
|
StaticCache,
|
||||||
)
|
)
|
||||||
from transformers.generation import GenerationMixin
|
from transformers.generation import GenerationMixin
|
||||||
@@ -31,6 +30,14 @@ from transformers.utils import (
|
|||||||
|
|
||||||
from .configuration_qwen2_5_vl import Qwen2_5_VLConfig, Qwen2_5_VLVisionConfig
|
from .configuration_qwen2_5_vl import Qwen2_5_VLConfig, Qwen2_5_VLVisionConfig
|
||||||
|
|
||||||
|
# TODO(Steven): SlidingWindowCache was removed in transformers v5. Define a placeholder so isinstance checks
|
||||||
|
# always return False (which is the correct behavior when no sliding window cache is in use).
|
||||||
|
class _SlidingWindowCachePlaceholder:
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
SlidingWindowCache = _SlidingWindowCachePlaceholder
|
||||||
|
|
||||||
if is_flash_attn_2_available():
|
if is_flash_attn_2_available():
|
||||||
from flash_attn import flash_attn_func, flash_attn_varlen_func
|
from flash_attn import flash_attn_func, flash_attn_varlen_func
|
||||||
from flash_attn.layers.rotary import apply_rotary_emb
|
from flash_attn.layers.rotary import apply_rotary_emb
|
||||||
|
|||||||
Reference in New Issue
Block a user