Update vision.py

2025-03-19 02:50:24 -07:00 · 2025-03-19 02:50:24 -07:00 · 2ef2724543
commit 2ef2724543
parent 95b4e83782
1 changed files with 1 additions and 0 deletions
--- a/unsloth/models/vision.py
+++ b/unsloth/models/vision.py
@ -201,6 +201,7 @@ def unsloth_base_fast_generate(
        kwargs["cache_implementation"] = cache_implementation
        kwargs["compile_config"] = _compile_config
    pass
+    print(kwargs)

    with torch.inference_mode(), autocaster:
        try: