mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-03-21 21:22:11 -03:00
refactor(downloader): update download_to_memory calls to include response headers
This commit is contained in:
@@ -94,7 +94,7 @@ class CivitaiClient:
|
||||
async def download_preview_image(self, image_url: str, save_path: str):
|
||||
try:
|
||||
downloader = await get_downloader()
|
||||
success, content = await downloader.download_to_memory(
|
||||
success, content, headers = await downloader.download_to_memory(
|
||||
image_url,
|
||||
use_auth=False # Preview images don't need auth
|
||||
)
|
||||
|
||||
@@ -463,7 +463,7 @@ class DownloadManager:
|
||||
|
||||
# Download the original image to temp path using downloader
|
||||
downloader = await get_downloader()
|
||||
success, content = await downloader.download_to_memory(
|
||||
success, content, headers = await downloader.download_to_memory(
|
||||
images[0]['url'],
|
||||
use_auth=False
|
||||
)
|
||||
|
||||
@@ -141,7 +141,7 @@ class ModelRouteUtils:
|
||||
else:
|
||||
# For images, download and then optimize to WebP using downloader
|
||||
downloader = await get_downloader()
|
||||
success, content = await downloader.download_to_memory(
|
||||
success, content, headers = await downloader.download_to_memory(
|
||||
first_preview['url'],
|
||||
use_auth=False
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user