Update handler.py
Browse files- handler.py +1 -0
handler.py
CHANGED
|
@@ -2,6 +2,7 @@ import torch
|
|
| 2 |
import unsloth
|
| 3 |
from transformers import AutoTokenizer, pipeline
|
| 4 |
from peft import AutoPeftModelForCausalLM
|
|
|
|
| 5 |
|
| 6 |
MODEL_NAME = "unsloth/Phi-4-unsloth-bnb-4bit" # Base model name (e.g., mistralai/Mistral-7B)
|
| 7 |
LORA_ADAPTER = "Machlovi/Safe_Phi4" # Your LoRA fine-tuned adapter
|
|
|
|
| 2 |
import unsloth
|
| 3 |
from transformers import AutoTokenizer, pipeline
|
| 4 |
from peft import AutoPeftModelForCausalLM
|
| 5 |
+
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 |
LORA_ADAPTER = "Machlovi/Safe_Phi4" # Your LoRA fine-tuned adapter
|