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