Update handler.py
Browse files- handler.py +2 -1
handler.py
CHANGED
|
@@ -6,7 +6,8 @@ from unsloth import FastLanguageModel # FastVisionModel for LLMs
|
|
| 6 |
|
| 7 |
MODEL_NAME = "unsloth/Phi-4-unsloth-bnb-4bit" # Base model name (e.g., mistralai/Mistral-7B)
|
| 8 |
model_id = "Machlovi/Safe_Phi4" # Your LoRA fine-tuned adapter
|
| 9 |
-
|
|
|
|
| 10 |
def load_model():
|
| 11 |
"""Loads the base model and LoRA adapter using Unsloth."""
|
| 12 |
print("Loading base model with Unsloth...")
|
|
|
|
| 6 |
|
| 7 |
MODEL_NAME = "unsloth/Phi-4-unsloth-bnb-4bit" # Base model name (e.g., mistralai/Mistral-7B)
|
| 8 |
model_id = "Machlovi/Safe_Phi4" # Your LoRA fine-tuned adapter
|
| 9 |
+
max_seq_length = 2048 # Choose any! We auto support RoPE Scaling internally!
|
| 10 |
+
load_in_4bit = True
|
| 11 |
def load_model():
|
| 12 |
"""Loads the base model and LoRA adapter using Unsloth."""
|
| 13 |
print("Loading base model with Unsloth...")
|