"...en/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "8690e8b9d6e68b3aec28f7f84374dd8256e79f18"
Unverified Commit 58f5f748 authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

[Tests] Fix paint by example (#4761)

* [Tests] Fix paint by example

* Update src/diffusers/pipelines/paint_by_example/image_encoder.py
parent 4f05058b
...@@ -23,9 +23,9 @@ logger = logging.get_logger(__name__) # pylint: disable=invalid-name ...@@ -23,9 +23,9 @@ logger = logging.get_logger(__name__) # pylint: disable=invalid-name
class PaintByExampleImageEncoder(CLIPPreTrainedModel): class PaintByExampleImageEncoder(CLIPPreTrainedModel):
def __init__(self, config, proj_size=768): def __init__(self, config, proj_size=None):
super().__init__(config) super().__init__(config)
self.proj_size = proj_size self.proj_size = proj_size or getattr(config, "projection_dim", 768)
self.model = CLIPVisionModel(config) self.model = CLIPVisionModel(config)
self.mapper = PaintByExampleMapper(config) self.mapper = PaintByExampleMapper(config)
......
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