Skip to content

GitLab

"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "a85b34e7fdc0a5fceb11aa0fa6199bd9afaca396"
Commit da8e87e2 authored by Patrick von Platen's avatar Patrick von Platen
Browse files

use real checkpoint

parent 43bbc781
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
debug_conversion.py
View file @ da8e87e2
...@@ -45,7 +45,6 @@ def current_codebase_conversion(): ...@@ -45,7 +45,6 @@ def current_codebase_conversion():
currently_converted_checkpoint = 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): def diff_between_checkpoints(ch_0, ch_1):
...@@ -84,3 +83,4 @@ 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) diff_between_checkpoints(currently_converted_checkpoint, checkpoint)
torch.save(checkpoint, "/path/to/checkpoint/")
tests/test_modeling_utils.py
View file @ da8e87e2
...@@ -1070,7 +1070,6 @@ class PipelineTesterMixin(unittest.TestCase): ...@@ -1070,7 +1070,6 @@ class PipelineTesterMixin(unittest.TestCase):
@slow @slow
def test_ldm_uncond(self): 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") ldm = LatentDiffusionUncondPipeline.from_pretrained("CompVis/latent-diffusion-celeba-256")
generator = torch.manual_seed(0) 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