Update Diffree_node.py

2024-08-06 15:33:28 +08:00
parent c6c8be272e
commit f687bd3f9b
1 changed files with 1 additions and 1 deletions
@@ -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