From f4173f3dda68cb658cb7f2b45427f7d9c70150d7 Mon Sep 17 00:00:00 2001 From: Cyril Vallez Date: Wed, 13 Aug 2025 19:54:07 +0200 Subject: [PATCH] fix --- src/transformers/models/janus/modeling_janus.py | 1 - src/transformers/models/janus/modular_janus.py | 1 - 2 files changed, 2 deletions(-) diff --git a/src/transformers/models/janus/modeling_janus.py b/src/transformers/models/janus/modeling_janus.py index 794514867a3c..4729a3677b4e 100644 --- a/src/transformers/models/janus/modeling_janus.py +++ b/src/transformers/models/janus/modeling_janus.py @@ -1360,7 +1360,6 @@ def generate( batch_size=batch_size * 2, # we should have at least a cache len of seq_len + num_image_tokens. max_cache_len=max(generation_config.max_length, num_image_tokens + seq_len), - device=device, model_kwargs=model_kwargs, ) diff --git a/src/transformers/models/janus/modular_janus.py b/src/transformers/models/janus/modular_janus.py index d90f409ee865..171a8acc814b 100644 --- a/src/transformers/models/janus/modular_janus.py +++ b/src/transformers/models/janus/modular_janus.py @@ -1220,7 +1220,6 @@ def generate( batch_size=batch_size * 2, # we should have at least a cache len of seq_len + num_image_tokens. max_cache_len=max(generation_config.max_length, num_image_tokens + seq_len), - device=device, model_kwargs=model_kwargs, )