AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

use real checkpoint

Browse Source
This commit is contained in:
Patrick von Platen
2022-07-15 19:13:39 +00:00
parent 43bbc78123
commit da8e87e201
2 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
debug_conversion.py
View File

@@ -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/")

1
tests/test_modeling_utils.py
View File

@@ -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)