riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-04 05:10:06 +06:00
Code Issues Actions 20 Packages Projects Releases Wiki Activity

break style

Browse Source
This commit is contained in:
ydshieh 2025-06-12 10:50:44 +02:00
parent 7c58336949
commit 1a58d86295
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/transformers/models/altclip/modeling_altclip.py
View File

@ -51,6 +51,15 @@ def clip_loss(similarity: torch.Tensor) -> torch.Tensor:
return (caption_loss + image_loss) / 2.0 return (caption_loss + image_loss) / 2.0
@dataclass @dataclass
# Copied from transformers.models.clip.modeling_clip.CLIPOutput with CLIP->AltCLIP # Copied from transformers.models.clip.modeling_clip.CLIPOutput with CLIP->AltCLIP
class AltCLIPOutput(ModelOutput): class AltCLIPOutput(ModelOutput):