Update finetune.py

loubnabnl · web-flow · commit cb299cfce41e · 2024-02-28T10:59:44.000+01:00
diff --git a/finetune.py b/finetune.py
@@ -86,7 +86,6 @@ def main(args):
         args.model_id,
         quantization_config=bnb_config,
         device_map={"": PartialState().process_index},
-        token=token,
         attention_dropout=args.attention_dropout,
     )
     print_trainable_parameters(model)

Original file line number	Diff line number	Diff line change
`@@ -86,7 +86,6 @@ def main(args):`
`86`	`86`	`args.model_id,`
`87`	`87`	`quantization_config=bnb_config,`
`88`	`88`	`device_map={"": PartialState().process_index},`
`89`		`- token=token,`
`90`	`89`	`attention_dropout=args.attention_dropout,`
`91`	`90`	`)`
`92`	`91`	`print_trainable_parameters(model)`