From 1d5e30c54c370917a130d83fca85ebdae6d69e39 Mon Sep 17 00:00:00 2001 From: tianxingwu Date: Thu, 21 Mar 2024 14:04:09 +0000 Subject: [PATCH] Fix SVD image input --- scripts/sampling/simple_video_sample.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/scripts/sampling/simple_video_sample.py b/scripts/sampling/simple_video_sample.py index 29a8b8581..dc8ba2100 100644 --- a/scripts/sampling/simple_video_sample.py +++ b/scripts/sampling/simple_video_sample.py @@ -164,6 +164,8 @@ def sample( with Image.open(input_img_path) as image: if image.mode == "RGBA": input_image = image.convert("RGB") + else: + input_image = image.copy() w, h = image.size if h % 64 != 0 or w % 64 != 0: