Update gaussian_diffusion.py
Browse files- gaussian_diffusion.py +1 -1
gaussian_diffusion.py
CHANGED
|
@@ -10,7 +10,7 @@ import numpy as np
|
|
| 10 |
import torch as th
|
| 11 |
import enum
|
| 12 |
|
| 13 |
-
from diffusion_utils import discretized_gaussian_log_likelihood, normal_kl
|
| 14 |
|
| 15 |
|
| 16 |
def mean_flat(tensor):
|
|
|
|
| 10 |
import torch as th
|
| 11 |
import enum
|
| 12 |
|
| 13 |
+
from .diffusion_utils import discretized_gaussian_log_likelihood, normal_kl
|
| 14 |
|
| 15 |
|
| 16 |
def mean_flat(tensor):
|