From 5c819f70d1b13b1b19bc479c3337598ffb18d911 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=A4=A9=E6=B6=AF=E5=8F=A4=E5=B7=B7?= Date: Sat, 23 Dec 2023 10:42:31 +0800 Subject: [PATCH] Update sttn_inpaint.py --- backend/inpaint/sttn_inpaint.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/backend/inpaint/sttn_inpaint.py b/backend/inpaint/sttn_inpaint.py index 834325f..5bbad84 100644 --- a/backend/inpaint/sttn_inpaint.py +++ b/backend/inpaint/sttn_inpaint.py @@ -4,7 +4,9 @@ import numpy as np import torch from torchvision import transforms from typing import List - +import sys +sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) +sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))) from backend import config from backend.inpaint.sttn.auto_sttn import InpaintGenerator from backend.inpaint.utils.sttn_utils import Stack, ToTorchFormatTensor