Update app.py
Browse files
app.py
CHANGED
@@ -15,7 +15,7 @@ tuned_adapter = "newronai/llama-2-7b-QLoRA-Trial1"
|
|
15 |
config = PeftConfig.from_pretrained(tuned_adapter)
|
16 |
model = AutoModelForCausalLM.from_pretrained(base_model,
|
17 |
use_cache="cache",
|
18 |
-
|
19 |
# quantization_config=bnb_config
|
20 |
)
|
21 |
|
|
|
15 |
config = PeftConfig.from_pretrained(tuned_adapter)
|
16 |
model = AutoModelForCausalLM.from_pretrained(base_model,
|
17 |
use_cache="cache",
|
18 |
+
load_in_8bit = True
|
19 |
# quantization_config=bnb_config
|
20 |
)
|
21 |
|