with torch.autocast('cuda'):

scripts/process_utils.py

@@ -317,7 +317,7 @@ def process_image(input_image, mode: str, weight1: float = 0.4, weight2: float =
 
     elif mode == "original":
         with torch.autocast('cuda'):
-            sotai_image = generate_sotai_image(refined_image, output_width, output_height)
+            sotai_image = generate_sotai_image(input_image, output_width, output_height)
         
         # スケッチ画像の生成
         input_image_np = np.array(input_image)