Skip to content

GitLab

Unverified Commit e5b728c2 authored by M-A's avatar M-A Committed by GitHub
Browse files

Fix import cycle (#153)

parent 8f1a95d8
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
photomaker/model_v2.py
View file @ e5b728c2
...@@ -6,7 +6,7 @@ import torch.nn as nn ...@@ -6,7 +6,7 @@ import torch.nn as nn
from transformers.models.clip.modeling_clip import CLIPVisionModelWithProjection from transformers.models.clip.modeling_clip import CLIPVisionModelWithProjection
from transformers.models.clip.configuration_clip import CLIPVisionConfig from transformers.models.clip.configuration_clip import CLIPVisionConfig
from . import FacePerceiverResampler from .resampler import FacePerceiverResampler
VISION_CONFIG_DICT = { VISION_CONFIG_DICT = {
"hidden_size": 1024, "hidden_size": 1024,
...@@ -153,4 +153,4 @@ class PhotoMakerIDEncoder_CLIPInsightfaceExtendtoken(CLIPVisionModelWithProjecti ...@@ -153,4 +153,4 @@ class PhotoMakerIDEncoder_CLIPInsightfaceExtendtoken(CLIPVisionModelWithProjecti
return updated_prompt_embeds return updated_prompt_embeds
if __name__ == "__main__": if __name__ == "__main__":
PhotoMakerIDEncoder_CLIPInsightfaceExtendtoken() PhotoMakerIDEncoder_CLIPInsightfaceExtendtoken()
\ No newline at end of file
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