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)