Image Feature Extraction
Transformers
Safetensors
csd
feature-extraction
style-similarity
custom_code
Instructions to use bigshanedogg/CSD with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use bigshanedogg/CSD with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-feature-extraction", model="bigshanedogg/CSD", trust_remote_code=True)# pip install -U transformers accelerate # Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("bigshanedogg/CSD", trust_remote_code=True, device_map="auto") - Notebooks
- Google Colab
- Kaggle
fix: call post_init() in CSDModel.__init__ (transformers>=5 all_tied_weights_keys)
Browse files- modeling_csd.py +4 -0
modeling_csd.py
CHANGED
|
@@ -130,6 +130,10 @@ class CSDModel(PreTrainedModel):
|
|
| 130 |
)
|
| 131 |
self.last_layer_style = nn.Parameter(torch.empty(config.width, config.embed_dim))
|
| 132 |
self.last_layer_content = nn.Parameter(torch.empty(config.width, config.embed_dim))
|
|
|
|
|
|
|
|
|
|
|
|
|
| 133 |
|
| 134 |
def forward(self, pixel_values: torch.Tensor) -> CSDOutput:
|
| 135 |
_feature = self.backbone(pixel_values)
|
|
|
|
| 130 |
)
|
| 131 |
self.last_layer_style = nn.Parameter(torch.empty(config.width, config.embed_dim))
|
| 132 |
self.last_layer_content = nn.Parameter(torch.empty(config.width, config.embed_dim))
|
| 133 |
+
# transformers>=5 sets weight-loading state (e.g. all_tied_weights_keys, read by
|
| 134 |
+
# from_pretrained) in PreTrainedModel.post_init(); call it so the load doesn't
|
| 135 |
+
# AttributeError. from_pretrained overwrites the freshly-inited weights afterward.
|
| 136 |
+
self.post_init()
|
| 137 |
|
| 138 |
def forward(self, pixel_values: torch.Tensor) -> CSDOutput:
|
| 139 |
_feature = self.backbone(pixel_values)
|