7 Commits

Author SHA1 Message Date
Alex "mcmonkey" Goodwin
57a91a2c1c test tweaks 2024-09-19 20:45:27 +09:00
Alex "mcmonkey" Goodwin
cfb5d6a532 also download to a tmp file not the live file
to avoid compounding errors from network failure
2024-09-19 20:06:53 +09:00
Alex "mcmonkey" Goodwin
c539ee55f4 fix & logging 2024-09-19 20:05:14 +09:00
Alex "mcmonkey" Goodwin
80a916777b actually, require folder_path and clean up some code 2024-09-19 19:47:27 +09:00
Alex "mcmonkey" Goodwin
618f4d848c initial folder path input on download api 2024-09-19 19:15:11 +09:00
Alex "mcmonkey" Goodwin
cbaffac4a5 (minor) format download_models.py 2024-09-19 19:00:16 +09:00
Robin Huang
3e52e0364c
Add model downloading endpoint. (#4248)
* Add model downloading endpoint.

* Move client session init to async function.

* Break up large function.

* Send "download_progress" as websocket event.

* Fixed

* Fixed.

* Use async mock.

* Move server set up to right before run call.

* Validate that model subdirectory cannot contain relative paths.

* Add download_model test checking for invalid paths.

* Remove DS_Store.

* Consolidate DownloadStatus and DownloadModelResult

* Add progress_interval as an optional parameter.

* Use tuple type from annotations.

* Use pydantic.

* Update comment.

* Revert "Use pydantic."

This reverts commit 7461e8eb0073add315c65c6f5e361f0891bffc7d.

* Add new line.

* Add newline EOF.

* Validate model filename as well.

* Add comment to not reply on internal.

* Restrict downloading to safetensor files only.
2024-08-13 15:48:52 -04:00