Update peft.py

Signed-off-by: Alexandros Koumparoulis <[email protected]>

examples/vlm/hf/peft.py

@@ -85,7 +85,7 @@ def fmt(sample):
     parser.add_argument('--accelerator', default='gpu', choices=['gpu'])
     parser.add_argument('--max-steps', type=int, default=100)
     parser.add_argument('--wandb-project', type=str, default=None)
-    parser.add_argument('--use-4bit', help="Load model in 4bit")
+    parser.add_argument('--use-4bit', help="Load model in 4bit", action="store_true")
     args = parser.parse_args()
 
     wandb = None