From 4b635bef125048d0e119c527e25662428c1561cf Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Tue, 18 Mar 2025 05:29:09 -0700 Subject: [PATCH] Update vision.py --- unsloth/models/vision.py | 1 + 1 file changed, 1 insertion(+) diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py index 53a873d168..8923491258 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -89,6 +89,7 @@ def unsloth_base_fast_generate( raise TypeError("Unsloth: You need to pass in input_ids to .generate!") assert(type(x) is torch.Tensor) bsz = x.shape[0] + print(kwargs) FastBaseModel.for_inference(self) dtype = _get_dtype(self.config.torch_dtype)