DummyObjects are impeding PyCharm

I’m subclassing StableDiffusionPipeline (because it seems like that’s the intended way to make a DiffusionPipeline that is still able to take advantage of StableDiffusionPipeline’s methods to enable attention slicing, decode latents, etc.), and I’ve noticed a distinct lack of help from the IDE when it comes to attributes and arguments that I know this class has.

I think that’s because when I hit “go to definition” of the StableDiffusionPipeline I’m using as a superclass, it brings me to this https://github.com/huggingface/diffusers/blob/0b7225e91852df668ce85a7f7a670c00272c9ed0/src/diffusers/utils/dummy_torch_and_transformers_objects.py#L127-L139

no defined arguments to the class __init__, no indication that it’s an ancestor of DiffusionPipeline, no __call__ method, none of the documentation or type hits that you’ve so diligently included in your code.

Should I be importing directly from the full path diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline instead of diffusers.StableDiffusionPipeline?

If so, please remove the diffusers.StableDiffusionPipeline stunt double so that I’m not presented with two options when autocompleting and left to probably pick the wrong one.

[anton-l]

Found the transformers solution, didn’t know it was already addressed there: https://github.com/huggingface/transformers/pull/16578 Assigning to myself as the only pycharm user on the team 😅

[sgugger]

Yes, the PR mentioned above fixes all editors normally, and will make them point to the dummy objects when the framework is not installed, and the real ones when it is 😃