tiny fix
Former-commit-id: f088e580d3bacd0eecd0c3bf17e928eb49832ba1
This commit is contained in:
@@ -145,7 +145,6 @@ def get_forbidden_modules(config: "PretrainedConfig", finetuning_args: "Finetuni
|
||||
|
||||
elif model_type == "minicpmv":
|
||||
if finetuning_args.freeze_vision_tower:
|
||||
print("******************", model_type)
|
||||
forbidden_modules.add("vpm")
|
||||
forbidden_modules.add("apm")
|
||||
forbidden_modules.add("resampler")
|
||||
|
||||
Reference in New Issue
Block a user