Machlovi commited on
Commit
f33435c
·
verified ·
1 Parent(s): 502a151

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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-4b-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,7 +41,7 @@ model, tokenizer = FastModel.from_pretrained(
41
  )
42
 
43
  # Load the fine-tuned LoRA adapter
44
- lora_model_name = "Machlovi/SafeLlama"
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()