diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py index acf999faff..f4e220ca8a 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -92,6 +92,7 @@ def unsloth_base_fast_generate( FastBaseModel.for_inference(self) dtype = _get_dtype(self.config.torch_dtype) + print(dtype) # Check if VLM is_vlm = any(