From da8e87e2018fc97ba9fd79fe8f51037e3aa19acf Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Fri, 15 Jul 2022 19:13:39 +0000 Subject: [PATCH] use real checkpoint --- debug_conversion.py | 2 +- tests/test_modeling_utils.py | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/debug_conversion.py b/debug_conversion.py index fa0d58d0ea..a8be53e822 100755 --- a/debug_conversion.py +++ b/debug_conversion.py @@ -45,7 +45,6 @@ def current_codebase_conversion(): currently_converted_checkpoint = current_codebase_conversion() -torch.save(currently_converted_checkpoint, 'currently_converted_checkpoint.pt') def diff_between_checkpoints(ch_0, ch_1): @@ -84,3 +83,4 @@ def diff_between_checkpoints(ch_0, ch_1): diff_between_checkpoints(currently_converted_checkpoint, checkpoint) +torch.save(checkpoint, "/path/to/checkpoint/") diff --git a/tests/test_modeling_utils.py b/tests/test_modeling_utils.py index 4aa30c645b..2bbc721a42 100755 --- a/tests/test_modeling_utils.py +++ b/tests/test_modeling_utils.py @@ -1070,7 +1070,6 @@ class PipelineTesterMixin(unittest.TestCase): @slow def test_ldm_uncond(self): -# ldm = LatentDiffusionUncondPipeline.from_pretrained("fusing/latent-diffusion-celeba-256", ldm=True) ldm = LatentDiffusionUncondPipeline.from_pretrained("CompVis/latent-diffusion-celeba-256") generator = torch.manual_seed(0)