Spaces:
Sleeping
Sleeping
update server.py
Browse files- download-model.py +3 -0
- 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 |
-
|
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()
|