kamp/IOPaint
1
0
Fork 0
Code Issues Pull Requests Releases 1 Activity

add back torch._C._jit_set_nvfuser_enabled(False)

Browse Source
This commit is contained in:
Qing 2024-01-08 23:12:32 +08:00
parent 0e6babea67
commit df5fd5ccf0
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
iopaint/api.py
View File

@ -10,6 +10,16 @@ import cv2
import numpy as np import numpy as np
import socketio import socketio
import torch 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 import uvicorn
from PIL import Image from PIL import Image
from fastapi import APIRouter, FastAPI, Request, UploadFile from fastapi import APIRouter, FastAPI, Request, UploadFile