Update README.md
Browse files
README.md
CHANGED
|
@@ -79,8 +79,8 @@ import warnings
|
|
| 79 |
|
| 80 |
# --- 3. Configuration ---
|
| 81 |
# Now both the base model and adapter are loaded from the iCIIT organization.
|
| 82 |
-
base_model_id = "iCIIT/
|
| 83 |
-
adapter_id = "iCIIT/
|
| 84 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 85 |
|
| 86 |
# --- 4. Load Model and Adapter ---
|
|
|
|
| 79 |
|
| 80 |
# --- 3. Configuration ---
|
| 81 |
# Now both the base model and adapter are loaded from the iCIIT organization.
|
| 82 |
+
base_model_id = "iCIIT/redqueenprotocol-sin-llama3.2-3B-model"
|
| 83 |
+
adapter_id = "iCIIT/redqueenprotocol-sin-llama3.2-3B-LoRA"
|
| 84 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 85 |
|
| 86 |
# --- 4. Load Model and Adapter ---
|