Update README.md
Browse files
README.md
CHANGED
@@ -51,7 +51,7 @@ make sure these packages are installed:
|
|
51 |
from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
|
52 |
import torch
|
53 |
import os
|
54 |
-
MODEL_PATH = "Neurai/
|
55 |
quantization_config = BitsAndBytesConfig(
|
56 |
load_in_4bit = True,
|
57 |
bnb_4bit_quant_type="nf4",
|
|
|
51 |
from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
|
52 |
import torch
|
53 |
import os
|
54 |
+
MODEL_PATH = "Neurai/NeuraOrcaGemma7b"
|
55 |
quantization_config = BitsAndBytesConfig(
|
56 |
load_in_4bit = True,
|
57 |
bnb_4bit_quant_type="nf4",
|