turnoff torchscript fuser feature
https://github.com/Sanster/lama-cleaner/issues/1
This commit is contained in:
9
main.py
9
main.py
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user