feat: clip model
This commit is contained in:
@@ -1,7 +1,9 @@
|
||||
from ._autoencoder import Autoencoder
|
||||
from ._clip import Clip
|
||||
from ._pipeline import Pipeline
|
||||
|
||||
__all__ = [
|
||||
"Pipeline",
|
||||
"Autoencoder",
|
||||
"Clip",
|
||||
]
|
||||
|
||||
@@ -0,0 +1,10 @@
|
||||
from dataclasses import dataclass
|
||||
|
||||
from transformers import CLIPTextModel, CLIPTokenizer
|
||||
|
||||
|
||||
@dataclass
|
||||
class Clip:
|
||||
tokenizer: CLIPTokenizer
|
||||
text_encoder: CLIPTextModel
|
||||
path: str
|
||||
Reference in New Issue
Block a user