[PPDiffusers] StableDiffusion Pipeline 升级 #390
Conversation
|
Thanks for your contribution! |
|
ok, 我看看 |
ppdiffusers/ppdiffusers/schedulers/scheduling_k_dpm_2_discrete.py
Outdated
Show resolved
Hide resolved
JunnYu
changed the title
|
不错，当前我这里测试的stable diffusion底下的单测都是正确的。 @co63oc 你好最后一个打错了，应该是这个文件，你可以再增加一下 https://github.com/huggingface/diffusers/blob/v0.24.0/tests/pipelines/stable_diffusion/test_stable_diffusion.py |
|
然后可以再测一下对应的slow的单测文件，export RUN_SLOW=True export RUN_NIGHTLY=True |
| >>> from ppdiffusers import StableDiffusionImg2ImgPipeline | ||
|
|
||
| >>> model_id_or_path = "runwayml/stable-diffusion-v1-5" | ||
| >>> pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_id_or_path, torch_dtype=paddle.float16) |
|
代码中涉及到的to(dtype)改成cast dtype，因为paddle的to是inplace操作，跟torch的操作不一样，可能未来某些时候会产生隐藏bug |
| else: | ||
| prompt_embeds_dtype = prompt_embeds.dtype | ||
|
|
||
| prompt_embeds = prompt_embeds.to(dtype=prompt_embeds_dtype) |
There was a problem hiding this comment.
已修改，test_stable_diffusion_inpaint_legacy.py 中部分使用to，如果使用cast有类型错误
| # Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline.encode_image | ||
| def encode_image(self, image, num_images_per_prompt): | ||
| # dtype = next(self.image_encoder.parameters()).dtype | ||
| dtype = next(self.image_encoder.parameters())[1].dtype |
There was a problem hiding this comment.
这里不能跟torch一样，因为paddle的parameters()不是迭代器拿到，而named_parameters()是迭代器，不过多了个name
已修改测试 test_stable_diffusion_img2img.py 和develop分支相同两项测试不通过 test_stable_diffusion.py test_stable_diffusion_inpaint_legacy.py |
|
OK, 你现在将最新的0.24.0（因为刚刚合入了一个新的关于scheduler和models单测的PR）合进去后，再重新跑一下题目中的4个文件对应的单测内容，跑完之后看看还有没有其他问题，如果没有就可以合入了（上面2个数值不通过的问题，可以忽略，这是nightly的） |
合并后SLOW测试不通过，比如 test_stable_diffusion_img2img.py::StableDiffusionImg2ImgPipelineSlowTests::test_stable_diffusion_img2img_ddim，pipeline生成的数据是随机的 |
|
好的，我这里试一下 |
|
OK,就这两个问题，可以合入了，这两个问题稍后再一起解决。 |
#377
完成1,2,3代码和单测，4和3单测相同
暂时测试可以，有其他问题再找时间修改
pytest test_stable_diffusion_img2img.py
pytest test_stable_diffusion_inpaint.py
pytest test_stable_diffusion_inpaint_legacy.py
