Skip to content

Commit da8e87e

Browse files
use real checkpoint
1 parent 43bbc78 commit da8e87e

2 files changed

Lines changed: 1 addition & 2 deletions

File tree

debug_conversion.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,6 @@ def current_codebase_conversion():
4545

4646

4747
currently_converted_checkpoint = current_codebase_conversion()
48-
torch.save(currently_converted_checkpoint, 'currently_converted_checkpoint.pt')
4948

5049

5150
def diff_between_checkpoints(ch_0, ch_1):
@@ -84,3 +83,4 @@ def diff_between_checkpoints(ch_0, ch_1):
8483

8584

8685
diff_between_checkpoints(currently_converted_checkpoint, checkpoint)
86+
torch.save(checkpoint, "/path/to/checkpoint/")

tests/test_modeling_utils.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1070,7 +1070,6 @@ def test_score_sde_vp_pipeline(self):
10701070

10711071
@slow
10721072
def test_ldm_uncond(self):
1073-
# ldm = LatentDiffusionUncondPipeline.from_pretrained("fusing/latent-diffusion-celeba-256", ldm=True)
10741073
ldm = LatentDiffusionUncondPipeline.from_pretrained("CompVis/latent-diffusion-celeba-256")
10751074

10761075
generator = torch.manual_seed(0)

0 commit comments

Comments
 (0)