From f687bd3f9b4ebedeb385db37e3ecac26d3e5b79f Mon Sep 17 00:00:00 2001
From: smthemex <138738845+smthemex@users.noreply.github.com>
Date: Tue, 6 Aug 2024 15:33:28 +0800
Subject: [PATCH] Update Diffree_node.py

---
 Diffree_node.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Diffree_node.py b/Diffree_node.py
index 485d467..4abe8e0 100644
--- a/Diffree_node.py
+++ b/Diffree_node.py
@@ -7,7 +7,7 @@ import sys
 import os
 from tqdm.auto import trange
 import einops
-import k_diffusion as K
+from . import k_diffusion as K
 import numpy as np
 import torch
 import torch.nn as nn