From f09d40cbef4a77099524f4af65678c033f1a86f7 Mon Sep 17 00:00:00 2001
From: Sanster <cwq1913@gmail.com>
Date: Fri, 18 Feb 2022 13:29:10 +0800
Subject: [PATCH] turnoff torchscript fuser feature

https://github.com/Sanster/lama-cleaner/issues/1
---
 README.md | 2 --
 main.py   | 9 +++++++++
 2 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index f0ae6cc..097d77a 100644
--- a/README.md
+++ b/README.md
@@ -53,8 +53,6 @@ docker run --gpus all -p 8080:8080 -e CACHE_DIR=/app/models -v $(pwd)/models:/ap
 
 Then open [http://localhost:8080](http://localhost:8080)
 
-https://www.buymeacoffee.com/Sanster
-
 ## Like My Work?
 <a href="https://www.buymeacoffee.com/Sanster"> 
   <img height="50em" src="https://cdn.buymeacoffee.com/buttons/v2/default-blue.png" alt="Sanster" />
diff --git a/main.py b/main.py
index f3b1b85..355c673 100644
--- a/main.py
+++ b/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