Hugging Face's logo

Spaces:
yunzi7
/
icebreaking
Runtime error

App Files Community
8
yunzi7 dongwook-chan commited on
Commit
1c1b26d
·
verified ·
1 Parent(s): 0f0b709

pr #8 (#8)

Browse files

- Change model to v5 (0a8fd6a48639dcbd008b9c49910b1b007864166f)


Co-authored-by: Dongwook Chang <dongwook-chan@users.noreply.huggingface.co>

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
  """