Spaces:
Sleeping
Sleeping
Update inference_script.py
Browse files- inference_script.py +1 -1
inference_script.py
CHANGED
|
@@ -4,7 +4,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
|
|
| 4 |
from peft import PeftModel
|
| 5 |
import os
|
| 6 |
|
| 7 |
-
ADAPTER_PATH = "
|
| 8 |
BASE_MODEL_NAME = "google/gemma-2b"
|
| 9 |
MAX_SEQ_LENGTH = 1024
|
| 10 |
MAX_NEW_TOKENS = 250
|
|
|
|
| 4 |
from peft import PeftModel
|
| 5 |
import os
|
| 6 |
|
| 7 |
+
ADAPTER_PATH = "KirubaLS/Outreach_model_6"
|
| 8 |
BASE_MODEL_NAME = "google/gemma-2b"
|
| 9 |
MAX_SEQ_LENGTH = 1024
|
| 10 |
MAX_NEW_TOKENS = 250
|