Fix imports.

Author: Adversarian

denoising_diffusion_pytorch/denoising_diffusion_pytorch.py

@@ -26,7 +26,7 @@
 from accelerate import Accelerator
 
 import numpy as np
-from fid_evaluation import FIDEvaluation
+from denoising_diffusion_pytorch.fid_evaluation import FIDEvaluation
 
 from denoising_diffusion_pytorch.version import __version__
 

denoising_diffusion_pytorch/fid_evaluation.py

@@ -9,7 +9,14 @@
 from torch.nn.functional import adaptive_avg_pool2d
 from tqdm.auto import tqdm
 
-from denoising_diffusion_pytorch import num_to_groups
+
+def num_to_groups(num, divisor):
+    groups = num // divisor
+    remainder = num % divisor
+    arr = [divisor] * groups
+    if remainder > 0:
+        arr.append(remainder)
+    return arr
 
 
 class FIDEvaluation: