Update main.py
Browse files
main.py
CHANGED
|
@@ -36,7 +36,7 @@ TOKENIZER_BIN = BASE_DIR / "tokenizer.bin"
|
|
| 36 |
|
| 37 |
# ⚠️ YAHAN APNA HUGGING FACE REPO ID DAALO ⚠️
|
| 38 |
# Example: "Sumeet/KVInfer-152M"
|
| 39 |
-
HF_REPO_ID
|
| 40 |
|
| 41 |
SYSTEM_TOKEN = "System:"
|
| 42 |
USER_TOKEN = "User:"
|
|
|
|
| 36 |
|
| 37 |
# ⚠️ YAHAN APNA HUGGING FACE REPO ID DAALO ⚠️
|
| 38 |
# Example: "Sumeet/KVInfer-152M"
|
| 39 |
+
HF_REPO_ID = "NOT-OMEGA/KVInfer-152M"
|
| 40 |
|
| 41 |
SYSTEM_TOKEN = "System:"
|
| 42 |
USER_TOKEN = "User:"
|