ros/LLaMA-Factory
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[misc] fix bug in constant (#7765)

Browse Source 
Co-authored-by: Sachin Beldona <sbeldona@cs.cmu.edu>
This commit is contained in:
Sachin Beldona
2025-04-21 10:06:31 -05:00
committed by GitHub
parent b605c20768
commit 3b2d3794a5
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/llamafactory/webui/control.py
View File

@@ -56,7 +56,7 @@ def can_quantize_to(quantization_method: str) -> "gr.Dropdown":
Inputs: top.quantization_method
Outputs: top.quantization_bit
"""
if quantization_method == QuantizationMethod.BITS_AND_BYTES.value:
if quantization_method == QuantizationMethod.BNB.value:
available_bits = ["none", "8", "4"]
elif quantization_method == QuantizationMethod.HQQ.value:
available_bits = ["none", "8", "6", "5", "4", "3", "2", "1"]