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
from transformers.models.clip.modeling_clip import CLIPVisionModelWithProjection
from transformers.models.clip.configuration_clip import CLIPVisionConfig
from . import FacePerceiverResampler
from .resampler import FacePerceiverResampler
VISION_CONFIG_DICT = {
"hidden_size": 1024,
......@@ -153,4 +153,4 @@ class PhotoMakerIDEncoder_CLIPInsightfaceExtendtoken(CLIPVisionModelWithProjecti
return updated_prompt_embeds
if __name__ == "__main__":
PhotoMakerIDEncoder_CLIPInsightfaceExtendtoken()
\ No newline at end of file
PhotoMakerIDEncoder_CLIPInsightfaceExtendtoken()
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