Machlovi commited on
Commit
5db5ca9
·
verified ·
1 Parent(s): 4123aac

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +2 -2
handler.py CHANGED
@@ -25,7 +25,7 @@ class EndpointHandler:
25
  "run_name": "Gemma12-MegaHateCat+",
26
  "model_id": "unsloth/gemma-3-12b-it-bnb-4bit",
27
  "hub_name": "Machlovi/Gemma3_12_MegaHateCatplus",
28
- "lora_adapter": f"{model_dir}/lora_adapter/gemma12"
29
  },
30
  "Qwen2.5": {
31
  "max_seq_length": 4096,
@@ -34,7 +34,7 @@ class EndpointHandler:
34
  "run_name": "Qwen2.5-MegaHateCat+",
35
  "model_id": "unsloth/Qwen2.5-7B",
36
  "hub_name": "Machlovi/Qwen2.5_MegaHateCatplus",
37
- "lora_adapter": f"{model_dir}/lora_adapter/qwen25"
38
  }
39
  }
40
 
 
25
  "run_name": "Gemma12-MegaHateCat+",
26
  "model_id": "unsloth/gemma-3-12b-it-bnb-4bit",
27
  "hub_name": "Machlovi/Gemma3_12_MegaHateCatplus",
28
+ "lora_adapter": "Machlovi/Gemma3_12_MegaHateCatplus"
29
  },
30
  "Qwen2.5": {
31
  "max_seq_length": 4096,
 
34
  "run_name": "Qwen2.5-MegaHateCat+",
35
  "model_id": "unsloth/Qwen2.5-7B",
36
  "hub_name": "Machlovi/Qwen2.5_MegaHateCatplus",
37
+ "lora_adapter": "Machlovi/Qwen2.5_MegaHateCatplus"
38
  }
39
  }
40