jchen8000 commited on
Commit
4639499
·
1 Parent(s): 5453328

update download model

Browse files
Files changed (1) hide show
  1. server.py +14 -5
server.py CHANGED
@@ -172,13 +172,22 @@ def create_interface():
172
  allowed_paths=["cache", "css", "extensions", "js"]
173
  )
174
 
175
- def download_model(model_name):
176
- logger.info("Downloading " + model_name)
177
- os.system(r"python3 download-model.py " + model_name)
178
- logger.info("Download completed " + model_name)
 
 
 
 
 
 
 
179
 
180
  if __name__ == "__main__":
181
- download_model("google/flan-t5-xl")
 
 
182
  logger.info("Starting Text generation web UI")
183
 
184
  do_cmd_flags_warnings()
 
172
  allowed_paths=["cache", "css", "extensions", "js"]
173
  )
174
 
175
+ def download_model(model_name, specific_file=None):
176
+ if specific_file:
177
+ logger.info("Downloading " + model_name + " - " + specific_file)
178
+ os.system(r"python3 download-model.py " + model_name + " --specific-file " + specific_file)
179
+ logger.info("Download completed " + model_name + " - " + specific_file)
180
+ else:
181
+ logger.info("Downloading " + model_name)
182
+ os.system(r"python3 download-model.py " + model_name)
183
+ logger.info("Download completed " + model_name)
184
+
185
+
186
 
187
  if __name__ == "__main__":
188
+ download_model("google/flan-t5-large")
189
+ download_model("MaziyarPanahi/Mistral-7B-Instruct-v0.3-GGUF", "Mistral-7B-Instruct-v0.3.Q4_K_M.gguf")
190
+
191
  logger.info("Starting Text generation web UI")
192
 
193
  do_cmd_flags_warnings()