Commit da8e87e2 authored by Patrick von Platen's avatar Patrick von Platen
Browse files

use real checkpoint

parent 43bbc781
......@@ -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/")
......@@ -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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment