Skip to content

Commit

Permalink
improve _get_features docs
Browse files Browse the repository at this point in the history
  • Loading branch information
rittik9 committed Jan 8, 2025
1 parent 7c93760 commit 00ee2e9
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/torchmetrics/functional/multimodal/clip_score.py
Original file line number Diff line number Diff line change
Expand Up @@ -97,7 +97,7 @@ def _get_features(
Args:
data: List of input data (images or text)
modality: Type of input data ("image" or "text")
modality: String indicating the type of input data (must be either "image" or "text")
device: Device to run the model on
model: CLIP model instance
processor: CLIP processor instance
Expand Down

0 comments on commit 00ee2e9

Please sign in to comment.