Merge pull request #129 from NoahSchiro/fp16-fix

BUG FIX: fp16 "TypeError: unflatten_dense_tensors(): argument 'tensors' (position 2) must be tuple of Tensors, not generator"

Author: unixpickle

improved_diffusion/fp16_util.py

@@ -65,7 +65,7 @@ def unflatten_master_params(model_params, master_params):
     """
     Unflatten the master parameters to look like model_params.
     """
-    return _unflatten_dense_tensors(master_params[0].detach(), model_params)
+    return _unflatten_dense_tensors(master_params[0].detach(), tuple(tensor for tensor in model_params))
 
 
 def zero_grad(model_params):