Fix bug when mixed precision training is set to off by use_fp16 == false

openai / consistency_models

Official repo for consistency models.

MIT License

6.17k stars 420 forks source link

Closed KeiKinn closed 4 months ago

KeiKinn commented 1 year ago

The original code in train_utils.py only supports for fp16 precision.
target_model_master_params will be set as list(target_model.parameters()) if use_fp16 == false.