Spaces:
Runtime error
Runtime error
changes made to app.py
Browse files
app.py
CHANGED
|
@@ -23,9 +23,12 @@ model.to(device)
|
|
| 23 |
# ===============================
|
| 24 |
# Load Evaluation Model (QwQ)
|
| 25 |
# ===============================
|
|
|
|
| 26 |
bnb_config = BitsAndBytesConfig(
|
| 27 |
-
|
| 28 |
-
|
|
|
|
|
|
|
| 29 |
)
|
| 30 |
|
| 31 |
qwq_model_id = "unsloth/QwQ-32B-unsloth-bnb-4bit"
|
|
|
|
| 23 |
# ===============================
|
| 24 |
# Load Evaluation Model (QwQ)
|
| 25 |
# ===============================
|
| 26 |
+
# Set 4-bit quantization configuration
|
| 27 |
bnb_config = BitsAndBytesConfig(
|
| 28 |
+
load_in_4bit=True,
|
| 29 |
+
bnb_4bit_compute_dtype=torch.bfloat16,
|
| 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"
|