423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-08-23 06:12:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

tiny fix

Browse Source 
Former-commit-id: 29c710da3a8984c5f6522a79368344544319ebce
This commit is contained in:
hiyouga 2024-06-27 00:36:04 +08:00
parent d2d9fa4abb
commit da80d41c78
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/llamafactory/webui/utils.py
View File

@ -61,11 +61,11 @@ def can_quantize_to(quantization_method: str) -> "gr.Dropdown":
Returns the available quantization bits. Returns the available quantization bits.
""" """
if quantization_method == QuantizationMethod.BITS_AND_BYTES.value: if quantization_method == QuantizationMethod.BITS_AND_BYTES.value:
return gr.Dropdown(choices=["none", "8", "4"]) return gr.Dropdown(choices=["none", "8", "4"], value="none")
elif quantization_method == QuantizationMethod.HQQ.value: elif quantization_method == QuantizationMethod.HQQ.value:
return gr.Dropdown(choices=["none", "8", "6", "5", "4", "3", "2", "1"]) return gr.Dropdown(choices=["none", "8", "6", "5", "4", "3", "2", "1"], value="none")
elif quantization_method == QuantizationMethod.EETQ.value: elif quantization_method == QuantizationMethod.EETQ.value:
return gr.Dropdown(choices=["none", "8"]) return gr.Dropdown(choices=["none", "8"], value="none")
def change_stage(training_stage: str = list(TRAINING_STAGES.keys())[0]) -> Tuple[List[str], bool]: def change_stage(training_stage: str = list(TRAINING_STAGES.keys())[0]) -> Tuple[List[str], bool]: