Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -32,10 +32,10 @@ import torch
|
|
| 32 |
|
| 33 |
# Create a configuration for quantization
|
| 34 |
quantization_config = BitsAndBytesConfig(
|
| 35 |
-
|
| 36 |
-
|
| 37 |
-
|
| 38 |
-
|
| 39 |
)
|
| 40 |
|
| 41 |
# Load the model and processor
|
|
|
|
| 32 |
|
| 33 |
# Create a configuration for quantization
|
| 34 |
quantization_config = BitsAndBytesConfig(
|
| 35 |
+
load_in_4bit=True,
|
| 36 |
+
bnb_4bit_compute_dtype="float16",
|
| 37 |
+
bnb_4bit_use_double_quant=True,
|
| 38 |
+
bnb_4bit_quant_type="nf4",
|
| 39 |
)
|
| 40 |
|
| 41 |
# Load the model and processor
|