jchen8000 commited on
Commit
a664870
·
1 Parent(s): 6e9ae82

update server.py

Browse files
Files changed (2) hide show
  1. download-model.py +3 -0
  2. server.py +1 -7
download-model.py CHANGED
@@ -265,6 +265,9 @@ class ModelDownloader:
265
  else:
266
  print(f"Downloading the model to {output_folder}")
267
 
 
 
 
268
  self.start_download_threads(links, output_folder, start_from_scratch=start_from_scratch, threads=threads)
269
 
270
  def check_model_files(self, model, branch, links, sha256, output_folder):
 
265
  else:
266
  print(f"Downloading the model to {output_folder}")
267
 
268
+ logger.info("Download links" + links)
269
+ logger.info("Download output_folder" + output_folder)
270
+
271
  self.start_download_threads(links, output_folder, start_from_scratch=start_from_scratch, threads=threads)
272
 
273
  def check_model_files(self, model, branch, links, sha256, output_folder):
server.py CHANGED
@@ -177,14 +177,8 @@ def download_model(model_name):
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"cd models")
183
- os.system(r"git clone "+ url)
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()
 
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("TheBloke/MythoMax-L2-13B-AWQ")
 
182
  logger.info("Starting Text generation web UI")
183
 
184
  do_cmd_flags_warnings()