From 9aeb8f886025e2a4741137e83fd9c6585f3d4afb 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:49:30 +0800 Subject: [PATCH] Update sttn_inpaint.py --- backend/inpaint/sttn_inpaint.py | 1 + 1 file changed, 1 insertion(+) diff --git a/backend/inpaint/sttn_inpaint.py b/backend/inpaint/sttn_inpaint.py index 5bbad84..9d2b862 100644 --- a/backend/inpaint/sttn_inpaint.py +++ b/backend/inpaint/sttn_inpaint.py @@ -5,6 +5,7 @@ import torch from torchvision import transforms from typing import List import sys +import os 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