diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py index 2436db4ff4..246e5d5671 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -608,7 +608,7 @@ class FastBaseModel: finetune_mlp_modules = finetune_mlp_modules, ) else: - assert(type(target_modules) in (list, tuple,)) + assert(type(target_modules) in (list, tuple, str,)) pass # Clear deleted GPU items