mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2025-08-03 20:22:49 +08:00
fix quantization is ""
Former-commit-id: ec2047b064bdb6a6e084aa268ee3de454f33dd59
This commit is contained in:
parent
802494e20a
commit
7b82ea708d
@ -112,7 +112,7 @@ class Runner:
|
||||
overwrite_cache=True,
|
||||
checkpoint_dir=checkpoint_dir,
|
||||
finetuning_type=finetuning_type,
|
||||
quantization_bit=int(quantization_bit) if quantization_bit != "None" else None,
|
||||
quantization_bit=int(quantization_bit) if quantization_bit & quantization_bit != "None" else None,
|
||||
template=template,
|
||||
system_prompt=system_prompt,
|
||||
dataset_dir=dataset_dir,
|
||||
@ -196,7 +196,7 @@ class Runner:
|
||||
predict_with_generate=True,
|
||||
checkpoint_dir=checkpoint_dir,
|
||||
finetuning_type=finetuning_type,
|
||||
quantization_bit=int(quantization_bit) if quantization_bit != "None" else None,
|
||||
quantization_bit=int(quantization_bit) if quantization_bit & quantization_bit != "None" else None,
|
||||
template=template,
|
||||
system_prompt=system_prompt,
|
||||
dataset_dir=dataset_dir,
|
||||
|
Loading…
x
Reference in New Issue
Block a user