update

app.py

@@ -102,7 +102,7 @@ print(NUM_THREADS)
 print("starting server ...")
 
 BASE_MODEL = "cyberagent/open-calm-7b"
-LORA_WEIGHTS_HF = "izumi-lab/stormy-7b-5ep"
+LORA_WEIGHTS_HF = "izumi-lab/stormy-7b-10ep"
 HF_TOKEN = os.environ.get("HF_TOKEN", None)
 DATASET_REPOSITORY = os.environ.get("DATASET_REPOSITORY", None)
 SLACK_WEBHOOK = os.environ.get("SLACK_WEBHOOK", None)

model_pull.py

@@ -4,7 +4,7 @@ from transformers import AutoModelForCausalLM
 from transformers import AutoTokenizer
 
 BASE_MODEL = "cyberagent/open-calm-7b"
-LORA_WEIGHTS = "izumi-lab/stormy-7b-5ep"
+LORA_WEIGHTS = "izumi-lab/stormy-7b-10ep"
 
 tokenizer = AutoTokenizer.from_pretrained(BASE_MODEL)
 model = AutoModelForCausalLM.from_pretrained(