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
|