masanorihirano commited on
Commit
b491e93
1 Parent(s): 16e584b
Files changed (2) hide show
  1. app.py +1 -1
  2. model_pull.py +1 -1
app.py CHANGED
@@ -102,7 +102,7 @@ print(NUM_THREADS)
102
  print("starting server ...")
103
 
104
  BASE_MODEL = "cyberagent/open-calm-7b"
105
- LORA_WEIGHTS_HF = "izumi-lab/stormy-7b-5ep"
106
  HF_TOKEN = os.environ.get("HF_TOKEN", None)
107
  DATASET_REPOSITORY = os.environ.get("DATASET_REPOSITORY", None)
108
  SLACK_WEBHOOK = os.environ.get("SLACK_WEBHOOK", None)
102
  print("starting server ...")
103
 
104
  BASE_MODEL = "cyberagent/open-calm-7b"
105
+ LORA_WEIGHTS_HF = "izumi-lab/stormy-7b-10ep"
106
  HF_TOKEN = os.environ.get("HF_TOKEN", None)
107
  DATASET_REPOSITORY = os.environ.get("DATASET_REPOSITORY", None)
108
  SLACK_WEBHOOK = os.environ.get("SLACK_WEBHOOK", None)
model_pull.py CHANGED
@@ -4,7 +4,7 @@ from transformers import AutoModelForCausalLM
4
  from transformers import AutoTokenizer
5
 
6
  BASE_MODEL = "cyberagent/open-calm-7b"
7
- LORA_WEIGHTS = "izumi-lab/stormy-7b-5ep"
8
 
9
  tokenizer = AutoTokenizer.from_pretrained(BASE_MODEL)
10
  model = AutoModelForCausalLM.from_pretrained(
4
  from transformers import AutoTokenizer
5
 
6
  BASE_MODEL = "cyberagent/open-calm-7b"
7
+ LORA_WEIGHTS = "izumi-lab/stormy-7b-10ep"
8
 
9
  tokenizer = AutoTokenizer.from_pretrained(BASE_MODEL)
10
  model = AutoModelForCausalLM.from_pretrained(