From 10f4f8ee2eb00135a09f908680b62e6aca4cef79 Mon Sep 17 00:00:00 2001
From: GF <149839341+gorillaframeai@users.noreply.github.com>
Date: Sun, 2 Feb 2025 19:23:25 +0300
Subject: [PATCH] Update gfrbmg2.py

---
 gfrbmg2.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/gfrbmg2.py b/gfrbmg2.py
index 93a0da7..6d41e68 100644
--- a/gfrbmg2.py
+++ b/gfrbmg2.py
@@ -1,6 +1,6 @@
 import sys
 import os
-sys.path.append(os.path.dirname(os.path.abspath(__file__)))
+
 
 import torch
 import folder_paths
@@ -11,6 +11,9 @@ from torchvision.transforms.functional import normalize
 import numpy as np
 import cv2
 import subprocess
+
+script_directory = os.path.dirname(os.path.realpath(__file__))
+sys.path.append(script_directory)
 import BEN2
 
 device = "cuda" if torch.cuda.is_available() else "cpu"