Update README.md
Browse files
README.md
CHANGED
|
@@ -33,7 +33,7 @@ from unsloth import FastModel
|
|
| 33 |
|
| 34 |
|
| 35 |
# Load the base model
|
| 36 |
-
base_model_name = "unsloth/gemma-3-
|
| 37 |
model, tokenizer = FastModel.from_pretrained(
|
| 38 |
model_name=base_model_name,
|
| 39 |
max_seq_length=2048, # Must match fine-tuning
|
|
@@ -41,7 +41,7 @@ model, tokenizer = FastModel.from_pretrained(
|
|
| 41 |
)
|
| 42 |
|
| 43 |
# Load the fine-tuned LoRA adapter
|
| 44 |
-
lora_model_name = "Machlovi/
|
| 45 |
model = PeftModel.from_pretrained(model, lora_model_name)
|
| 46 |
|
| 47 |
model.eval()
|
|
|
|
| 33 |
|
| 34 |
|
| 35 |
# Load the base model
|
| 36 |
+
base_model_name = "unsloth/gemma-3-12b-it-unsloth-bnb-4bit",",
|
| 37 |
model, tokenizer = FastModel.from_pretrained(
|
| 38 |
model_name=base_model_name,
|
| 39 |
max_seq_length=2048, # Must match fine-tuning
|
|
|
|
| 41 |
)
|
| 42 |
|
| 43 |
# Load the fine-tuned LoRA adapter
|
| 44 |
+
lora_model_name = "Machlovi/GGuard"
|
| 45 |
model = PeftModel.from_pretrained(model, lora_model_name)
|
| 46 |
|
| 47 |
model.eval()
|