Spaces:
Paused
Paused
without bnb_config
Browse files
app.py
CHANGED
|
@@ -41,12 +41,12 @@ books_db_client = books_db.as_retriever()
|
|
| 41 |
# Initialize the model and tokenizer
|
| 42 |
model_name = "stabilityai/stablelm-zephyr-3b"
|
| 43 |
|
| 44 |
-
bnb_config = transformers.BitsAndBytesConfig(
|
| 45 |
-
|
| 46 |
-
|
| 47 |
-
|
| 48 |
-
|
| 49 |
-
)
|
| 50 |
|
| 51 |
model_config = transformers.AutoConfig.from_pretrained(model_name, max_new_tokens=1024)
|
| 52 |
model = transformers.AutoModelForCausalLM.from_pretrained(
|
|
|
|
| 41 |
# Initialize the model and tokenizer
|
| 42 |
model_name = "stabilityai/stablelm-zephyr-3b"
|
| 43 |
|
| 44 |
+
# bnb_config = transformers.BitsAndBytesConfig(
|
| 45 |
+
# load_in_4bit=True,
|
| 46 |
+
# bnb_4bit_quant_type='nf4',
|
| 47 |
+
# bnb_4bit_use_double_quant=True,
|
| 48 |
+
# bnb_4bit_compute_dtype=torch.bfloat16
|
| 49 |
+
# )
|
| 50 |
|
| 51 |
model_config = transformers.AutoConfig.from_pretrained(model_name, max_new_tokens=1024)
|
| 52 |
model = transformers.AutoModelForCausalLM.from_pretrained(
|