Hugging Face's logo

Spaces:
yunzi7
/
icebreaking
Runtime error

App Files Community
8
dongwook-chan commited on
Commit
0a8fd6a
•
1 Parent(s): 0f0b709

Change model to v5

Browse files
Files changed (1) hide show
  1. ice_breaking_challenge/models/model_loader.py +2 -1
ice_breaking_challenge/models/model_loader.py CHANGED
@@ -1,7 +1,8 @@
1
  import keras_nlp
2
 
3
  MODEL_NAME = "gemma2_instruct_2b_en"
4
- LORA_WEIGHT_PATH = "ice_breaking_challenge/models/gemma2_it_2b_icebreaking.lora.h5"
 
5
 
6
  def load_model_with_lora(model_name:str = MODEL_NAME, lora_weight_path: str = LORA_WEIGHT_PATH):
7
  """
 
1
  import keras_nlp
2
 
3
  MODEL_NAME = "gemma2_instruct_2b_en"
4
+ #LORA_WEIGHT_PATH = "ice_breaking_challenge/models/gemma2_it_2b_icebreaking_quiz_v2_3.lora.h5"
5
+ LORA_WEIGHT_PATH = "ice_breaking_challenge/models/gemma2_it_2b_icebreaking_quiz_v2_5.lora.h5"
6
 
7
  def load_model_with_lora(model_name:str = MODEL_NAME, lora_weight_path: str = LORA_WEIGHT_PATH):
8
  """