From ea281400f8dd158752a006626ca28fc7fd1248dc Mon Sep 17 00:00:00 2001 From: Pedro Cuenca Date: Mon, 30 Jan 2023 16:57:01 +0100 Subject: [PATCH 1/2] Don't copy when unwrapping model. Otherwise an exception is raised when using fp16. --- examples/unconditional_image_generation/train_unconditional.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/unconditional_image_generation/train_unconditional.py b/examples/unconditional_image_generation/train_unconditional.py index 3b0177402c4e..b21334a8c868 100644 --- a/examples/unconditional_image_generation/train_unconditional.py +++ b/examples/unconditional_image_generation/train_unconditional.py @@ -530,7 +530,7 @@ def transforms(examples): # Generate sample images for visual inspection if accelerator.is_main_process: if epoch % args.save_images_epochs == 0 or epoch == args.num_epochs - 1: - unet = copy.deepcopy(accelerator.unwrap_model(model)) + unet = accelerator.unwrap_model(model) if args.use_ema: ema_model.copy_to(unet.parameters()) pipeline = DDPMPipeline( From 27df54c10714df86d3ebf5f71f3e28635f6854b2 Mon Sep 17 00:00:00 2001 From: Pedro Cuenca Date: Mon, 30 Jan 2023 17:57:54 +0100 Subject: [PATCH 2/2] Remove unused import --- examples/unconditional_image_generation/train_unconditional.py | 1 - 1 file changed, 1 deletion(-) diff --git a/examples/unconditional_image_generation/train_unconditional.py b/examples/unconditional_image_generation/train_unconditional.py index b21334a8c868..3ca92717f8fa 100644 --- a/examples/unconditional_image_generation/train_unconditional.py +++ b/examples/unconditional_image_generation/train_unconditional.py @@ -1,5 +1,4 @@ import argparse -import copy import inspect import logging import math