This commit is contained in:
hiyouga committed 2023-10-19 15:52:24 +08:00
1 parent 1611fad6bc
commit 6496a99b7d
3 files changed
+5 -4

No files matched your search

+3
View File
@@ -240,4 +240,7 @@ def load_model_and_tokenizer(
trainable_params, all_param, 100 * trainable_params / all_param
))
if not is_trainable:
logger.info("This IS expected that the trainable params is 0 if you are using model for inference only.")
return model, tokenizer