feat: clip model

This commit is contained in:
Limitex
2025-11-02 23:05:21 +09:00
parent c87630816b
commit f0f22cab08
6 changed files with 78 additions and 2 deletions
+2
View File
@@ -1,7 +1,9 @@
from ._autoencoder import Autoencoder
from ._clip import Clip
from ._pipeline import Pipeline
__all__ = [
"Pipeline",
"Autoencoder",
"Clip",
]
+10
View File
@@ -0,0 +1,10 @@
from dataclasses import dataclass
from transformers import CLIPTextModel, CLIPTokenizer
@dataclass
class Clip:
tokenizer: CLIPTokenizer
text_encoder: CLIPTextModel
path: str