diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py index e3fa946b7a..7993a9a48e 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -137,7 +137,7 @@ def unsloth_base_fast_generate( try: kwargs["pixel_values"] = kwargs["pixel_values"].to(dtype) except: pass - print(kwargs) + print(kwargs.keys()) # Mixed precision autocast if os.environ.get("UNSLOTH_FORCE_FLOAT32", "0") == "1": dtype = torch.float32