fix_cuda0_mem_usage_error (#770)

siliconflow · Mar 28, 2024 · 089489a · 089489a
1 parent b89dd49
commit 089489a
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/onediff_diffusers_extensions/onediffx/utils/patch_image_processor.py b/onediff_diffusers_extensions/onediffx/utils/patch_image_processor.py
@@ -82,7 +82,6 @@ def pt_to_numpy(images: torch.FloatTensor) -> np.ndarray:
     return _pt_to_numpy_pre(images).numpy()
 
 
-@torch.jit.script
 def _pt_to_pil_pre(images):
     return (
         images.permute(0, 2, 3, 1)