support system column #1765

This commit is contained in:
hiyouga committed 2023-12-12 19:45:59 +08:00
1 parent d5b2c57a35
commit 0a9c6e0146
15 files changed
+45 -396120

No files matched your search

+1 -1
View File
@@ -217,7 +217,7 @@ def load_model_and_tokenizer(
# Prepare model for inference
if not is_trainable:
model.requires_grad_(False) # fix all model params
model = model.to(model_args.compute_dtype) if model_args.quantization_bit is None else model
model = model.to(model_args.compute_dtype) if not getattr(model, "quantization_method", None) else model
model.eval()
else:
model.train()