Skip to content

GitLab

Commit ab00f5d3 authored by anton-l's avatar anton-l
Browse files

Update model names for CompVis and google

parent 3f0b44b3
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
tests/test_modeling_utils.py
View file @ ab00f5d3
......@@ -969,7 +969,7 @@ class PipelineTesterMixin(unittest.TestCase):
@slow
def test_ddim_lsun(self):
model_id = "google/ddpm-lsun-bedroom-ema"
model_id = "google/ddpm-ema-bedroom-256"
unet = UNetUnconditionalModel.from_pretrained(model_id)
scheduler = DDIMScheduler.from_config(model_id)
......@@ -1028,7 +1028,7 @@ class PipelineTesterMixin(unittest.TestCase):
@slow
def test_ldm_text2img(self):
ldm = LatentDiffusionPipeline.from_pretrained("CompVis/latent-diffusion-text2im-large")
ldm = LatentDiffusionPipeline.from_pretrained("CompVis/ldm-text2im-large")
prompt = "A painting of a squirrel eating a burger"
generator = torch.manual_seed(0)
......@@ -1042,7 +1042,7 @@ class PipelineTesterMixin(unittest.TestCase):
@slow
def test_ldm_text2img_fast(self):
ldm = LatentDiffusionPipeline.from_pretrained("CompVis/latent-diffusion-text2im-large")
ldm = LatentDiffusionPipeline.from_pretrained("CompVis/ldm-text2im-large")
prompt = "A painting of a squirrel eating a burger"
generator = torch.manual_seed(0)
......@@ -1121,7 +1121,7 @@ class PipelineTesterMixin(unittest.TestCase):
@slow
def test_ldm_uncond(self):
ldm = LatentDiffusionUncondPipeline.from_pretrained("CompVis/latent-diffusion-celeba-256")
ldm = LatentDiffusionUncondPipeline.from_pretrained("CompVis/ldm-celebahq-256")
generator = torch.manual_seed(0)
image = ldm(generator=generator, num_inference_steps=5)["sample"]
......
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