Spaces:
Sleeping
Sleeping
update download model
Browse files
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 |
-
|
177 |
-
|
178 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
179 |
|
180 |
if __name__ == "__main__":
|
181 |
-
download_model("google/flan-t5-
|
|
|
|
|
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()
|