turnoff torchscript fuser feature

https://github.com/Sanster/lama-cleaner/issues/1
This commit is contained in:
Sanster
2022-02-18 13:29:10 +08:00
parent 799ab7607b
commit f09d40cbef
2 changed files with 9 additions and 2 deletions

View File

@@ -11,6 +11,15 @@ from typing import Union
import cv2
import numpy as np
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
from flask import Flask, request, send_file
from flask_cors import CORS