Spaces:
Running
Running
changes made in app.py
Browse files
app.py
CHANGED
@@ -23,12 +23,9 @@ model.to(device)
|
|
23 |
# ===============================
|
24 |
# Load Evaluation Model (QwQ)
|
25 |
# ===============================
|
26 |
-
# Set 4-bit quantization configuration
|
27 |
bnb_config = BitsAndBytesConfig(
|
28 |
-
|
29 |
-
|
30 |
-
bnb_4bit_use_double_quant=True,
|
31 |
-
bnb_4bit_quant_type="nf4"
|
32 |
)
|
33 |
|
34 |
qwq_model_id = "unsloth/QwQ-32B-unsloth-bnb-4bit"
|
|
|
23 |
# ===============================
|
24 |
# Load Evaluation Model (QwQ)
|
25 |
# ===============================
|
|
|
26 |
bnb_config = BitsAndBytesConfig(
|
27 |
+
load_in_8bit=True,
|
28 |
+
llm_int8_enable_fp32_cpu_offload=True,
|
|
|
|
|
29 |
)
|
30 |
|
31 |
qwq_model_id = "unsloth/QwQ-32B-unsloth-bnb-4bit"
|