some cleanup

dalle2_pytorch/trainer.py

@@ -238,7 +238,7 @@ class EMA(nn.Module):
             ma_buffer.data.copy_(current_buffer.data)
 
     def get_current_decay(self):
-        epoch = clamp(self.step.item() - self.update_after_step - 1, min = 0)
+        epoch = clamp(self.step.item() - self.update_after_step - 1, min_value = 0)
         value = 1 - (1 + epoch / self.inv_gamma) ** - self.power
 
         if epoch <= 0:

dalle2_pytorch/version.py

@@ -1 +1 @@
-__version__ = '0.6.14'
+__version__ = '0.6.15'