Use different requestor for downloaders

This commit is contained in:
Nicolas Mowen 2024-10-12 09:26:13 -06:00
parent 6e332bbdf8
commit f58a2b8eba

View File

@ -44,7 +44,6 @@ class ModelDownloader:
download_path: str, download_path: str,
file_names: List[str], file_names: List[str],
download_func: Callable[[str], None], download_func: Callable[[str], None],
requestor: InterProcessRequestor,
silent: bool = False, silent: bool = False,
): ):
self.model_name = model_name self.model_name = model_name
@ -52,7 +51,7 @@ class ModelDownloader:
self.file_names = file_names self.file_names = file_names
self.download_func = download_func self.download_func = download_func
self.silent = silent self.silent = silent
self.requestor = requestor self.requestor = InterProcessRequestor()
self.download_thread = None self.download_thread = None
self.download_complete = threading.Event() self.download_complete = threading.Event()