Update app.py
Browse files
app.py
CHANGED
|
@@ -26,7 +26,7 @@ retriever = vectorstore.as_retriever()
|
|
| 26 |
# 2. Load Model & Tokenizer in 4-bit
|
| 27 |
# ==========================================
|
| 28 |
print("Loading Model...")
|
| 29 |
-
model_id = "
|
| 30 |
|
| 31 |
# Load in 4-bit to fit perfectly inside a T4 GPU Space
|
| 32 |
bnb_config = BitsAndBytesConfig(
|
|
|
|
| 26 |
# 2. Load Model & Tokenizer in 4-bit
|
| 27 |
# ==========================================
|
| 28 |
print("Loading Model...")
|
| 29 |
+
model_id = "anirudh248/llama3-upf-generator" # UPDATE THIS if your username is different
|
| 30 |
|
| 31 |
# Load in 4-bit to fit perfectly inside a T4 GPU Space
|
| 32 |
bnb_config = BitsAndBytesConfig(
|