Machlovi
/

Safe_Phi4_Full2

Model card Files Files and versions

Machlovi commited on Apr 4, 2025

Commit

b2d76ae

·

verified ·

1 Parent(s): a40b704

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -5,7 +5,7 @@ from peft import AutoPeftModelForCausalLM
 from unsloth import FastLanguageModel  # FastVisionModel for LLMs
 MODEL_NAME = "unsloth/Phi-4-unsloth-bnb-4bit"  # Base model name (e.g., mistralai/Mistral-7B)
-LORA_ADAPTER =  "Machlovi/Safe_Phi4" # Your LoRA fine-tuned adapter
 def load_model():
     """Loads the base model and LoRA adapter using Unsloth."""

 from unsloth import FastLanguageModel  # FastVisionModel for LLMs
 MODEL_NAME = "unsloth/Phi-4-unsloth-bnb-4bit"  # Base model name (e.g., mistralai/Mistral-7B)
+model_id =  "Machlovi/Safe_Phi4" # Your LoRA fine-tuned adapter
 def load_model():
     """Loads the base model and LoRA adapter using Unsloth."""