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

Update model names for CompVis and google

Browse Source
This commit is contained in:
anton-l
2022-07-19 15:13:22 +02:00
parent 3f0b44b322
commit ab00f5d3e1
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/test_modeling_utils.py
View File

@@ -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"]