Hugging Face's logo

Spaces:
yunzi7
/
icebreaking
Runtime error

App Files Community
8

pr #8

#8
by dongwook-chan - opened
base: refs/heads/main
←
from: refs/pr/8
Discussion Files changed
+2
-1
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
  """