Skip to content

Commit fd59db2

Browse files
authored
Merge pull request #228 from Adversarian/fix-unbound-fid
Bind `self.calculate_fid` for the case when FID is not requested.
2 parents f76f0fc + cc7a439 commit fd59db2

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

denoising_diffusion_pytorch/denoising_diffusion_pytorch.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -912,8 +912,8 @@ def __init__(
912912

913913
# FID-score computation
914914

915+
self.calculate_fid = calculate_fid
915916
if calculate_fid:
916-
self.calculate_fid = True
917917
if not self.model.is_ddim_sampling:
918918
self.accelerator.print(
919919
"WARNING: Robust FID computation requires a lot of generated samples and can therefore be very time consuming."\
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
__version__ = '1.8.0'
1+
__version__ = '1.8.1'

0 commit comments

Comments
 (0)