Update app.py
Browse files
app.py
CHANGED
|
@@ -39,7 +39,7 @@ def load_llama_model():
|
|
| 39 |
|
| 40 |
# Load Llama 3.2 model
|
| 41 |
MODEL_NAME = "meta-llama/Llama-3.2-1B-Instruct-QLORA_INT4_EO8"
|
| 42 |
-
tokenizer, model = load_llama_model(
|
| 43 |
|
| 44 |
# Load Llama Guard for content moderation
|
| 45 |
LLAMA_GUARD_NAME = "meta-llama/Llama-Guard-3-1B-INT4"
|
|
|
|
| 39 |
|
| 40 |
# Load Llama 3.2 model
|
| 41 |
MODEL_NAME = "meta-llama/Llama-3.2-1B-Instruct-QLORA_INT4_EO8"
|
| 42 |
+
tokenizer, model = load_llama_model()
|
| 43 |
|
| 44 |
# Load Llama Guard for content moderation
|
| 45 |
LLAMA_GUARD_NAME = "meta-llama/Llama-Guard-3-1B-INT4"
|