From df5fd5ccf031c17f1434a5407a79e50c41baa465 Mon Sep 17 00:00:00 2001 From: Qing Date: Mon, 8 Jan 2024 23:12:32 +0800 Subject: [PATCH] add back torch._C._jit_set_nvfuser_enabled(False) --- iopaint/api.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/iopaint/api.py b/iopaint/api.py index bd924fd..e8227e0 100644 --- a/iopaint/api.py +++ b/iopaint/api.py @@ -10,6 +10,16 @@ import cv2 import numpy as np import socketio import torch + +try: + torch._C._jit_override_can_fuse_on_cpu(False) + torch._C._jit_override_can_fuse_on_gpu(False) + torch._C._jit_set_texpr_fuser_enabled(False) + torch._C._jit_set_nvfuser_enabled(False) +except: + pass + + import uvicorn from PIL import Image from fastapi import APIRouter, FastAPI, Request, UploadFile