jchen8000 commited on
Commit
4ce74b5
·
1 Parent(s): 349ffb4

update server.py

Browse files
Files changed (1) hide show
  1. server.py +6 -1
server.py CHANGED
@@ -177,9 +177,14 @@ def download_model(model_name):
177
  os.system(r"python3 download-model.py " + model_name)
178
  logger.info("Download completed " + model_name)
179
 
 
 
 
 
180
 
181
  if __name__ == "__main__":
182
- download_model("TheBloke/MythoMax-L2-13B-AWQ")
 
183
  logger.info("Starting Text generation web UI")
184
 
185
  do_cmd_flags_warnings()
 
177
  os.system(r"python3 download-model.py " + model_name)
178
  logger.info("Download completed " + model_name)
179
 
180
+ def download_model_from_url(url):
181
+ logger.info("Downloading from" + url)
182
+ os.system(r"git clone "+ url r"/home/user/app/text-generation-webui/models")
183
+
184
 
185
  if __name__ == "__main__":
186
+ # download_model("TheBloke/MythoMax-L2-13B-AWQ")
187
+ download_model_from_url("https://huggingface.co/mistralai/Mistral-7B-Instruct-v0.3")
188
  logger.info("Starting Text generation web UI")
189
 
190
  do_cmd_flags_warnings()