Merge remote-tracking branch 'origin/main' into feat/ui/nodes-api

This commit is contained in:
psychedelicious 2023-04-05 18:05:23 +10:00
commit abc10a115d
2 changed files with 2 additions and 5 deletions

View File

@ -456,7 +456,7 @@ def get_torch_source() -> (Union[str, None],str):
optional_modules = None
if OS == "Linux":
if device == "rocm":
url = "https://download.pytorch.org/whl/rocm5.2"
url = "https://download.pytorch.org/whl/rocm5.4.2"
elif device == "cpu":
url = "https://download.pytorch.org/whl/cpu"

View File

@ -17,11 +17,8 @@ def save_thumbnail(
if os.path.exists(thumbnail_path):
return thumbnail_path
thumbnail_width = size
thumbnail_height = round(size * (image.height / image.width))
image_copy = image.copy()
image_copy.thumbnail(size=(thumbnail_width, thumbnail_height))
image_copy.thumbnail(size=(size, size))
image_copy.save(thumbnail_path, "WEBP")