Machlovi commited on
Commit
10250a3
·
verified ·
1 Parent(s): b2d76ae

Update handler.py

Browse files
Files changed (1) hide show
  1. 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...")