jlov7 commited on
Commit
773373d
Β·
1 Parent(s): 5c9ebbd

fix: prioritize Hub model loading for Spaces deployment

Browse files
Files changed (1) hide show
  1. test_constrained_model.py +8 -8
test_constrained_model.py CHANGED
@@ -35,20 +35,20 @@ def load_trained_model():
35
  low_cpu_mem_usage=True # Reduce memory usage during loading
36
  )
37
 
38
- # Try to load fine-tuned adapter - local first, then Hub
39
  try:
40
- print("πŸ”„ Attempting to load fine-tuned adapter locally...")
41
  from peft import PeftModel
42
- model = PeftModel.from_pretrained(model, "./smollm3_robust")
43
  model = model.merge_and_unload()
44
- print("βœ… Fine-tuned model loaded successfully from local files!")
45
  except Exception as e:
46
  try:
47
- print(f"⚠️ Local adapter failed: {e}")
48
- print("πŸ”„ Trying Hugging Face Hub...")
49
- model = PeftModel.from_pretrained(model, "jlov7/SmolLM3-Function-Calling-LoRA")
50
  model = model.merge_and_unload()
51
- print("βœ… Fine-tuned model loaded successfully from Hub!")
52
  except Exception as e2:
53
  print(f"⚠️ Could not load fine-tuned adapter: {e2}")
54
  print("πŸ”§ Using base model with optimized prompting")
 
35
  low_cpu_mem_usage=True # Reduce memory usage during loading
36
  )
37
 
38
+ # Try to load fine-tuned adapter - Hub first for Spaces deployment
39
  try:
40
+ print("πŸ”„ Loading fine-tuned adapter from Hugging Face Hub...")
41
  from peft import PeftModel
42
+ model = PeftModel.from_pretrained(model, "jlov7/SmolLM3-Function-Calling-LoRA")
43
  model = model.merge_and_unload()
44
+ print("βœ… Fine-tuned model loaded successfully from Hub!")
45
  except Exception as e:
46
  try:
47
+ print(f"⚠️ Hub adapter failed: {e}")
48
+ print("πŸ”„ Trying local backup...")
49
+ model = PeftModel.from_pretrained(model, "./smollm3_robust")
50
  model = model.merge_and_unload()
51
+ print("βœ… Fine-tuned model loaded successfully from local files!")
52
  except Exception as e2:
53
  print(f"⚠️ Could not load fine-tuned adapter: {e2}")
54
  print("πŸ”§ Using base model with optimized prompting")