Update finetune.py

linhduongtuan · web-flow · commit 653bfda58362 · 2023-03-28T20:33:48.000+07:00
diff --git a/finetune.py b/finetune.py
@@ -71,7 +71,7 @@ def train(
     )
     assert (
         base_model
-    ), "Please specify a --base_model, e.g. --base_model='LinhDuong/doctorwithbloom'"
+    ), "Please specify a --base_model, e.g. --base_model='bigscience/bloomz-7b1-mt'"
     gradient_accumulation_steps = batch_size // micro_batch_size
 
     device_map = "auto"

Original file line number	Diff line number	Diff line change
`@@ -71,7 +71,7 @@ def train(`
`71`	`71`	`)`
`72`	`72`	`assert (`
`73`	`73`	`base_model`
`74`		`- ), "Please specify a --base_model, e.g. --base_model='LinhDuong/doctorwithbloom'"`
	`74`	`+ ), "Please specify a --base_model, e.g. --base_model='bigscience/bloomz-7b1-mt'"`
`75`	`75`	`gradient_accumulation_steps = batch_size // micro_batch_size`
`76`	`76`
`77`	`77`	`device_map = "auto"`