diff --git a/rig-core/src/embeddings/embedding.rs b/rig-core/src/embeddings/embedding.rs
index 09ce7e0..4b9a32a 100644
--- a/rig-core/src/embeddings/embedding.rs
+++ b/rig-core/src/embeddings/embedding.rs
@@ -60,21 +60,21 @@ pub trait EmbeddingModel: Clone + Sync + Send {
     }
 }
 
-/// Trait for embedding models that can generate embeddings for documents.
+/// Trait for embedding models that can generate embeddings for images.
 pub trait ImageEmbeddingModel: Clone + Sync + Send {
-    /// The maximum number of documents that can be embedded in a single request.
+    /// The maximum number of images that can be embedded in a single request.
     const MAX_DOCUMENTS: usize;
 
     /// The number of dimensions in the embedding vector.
     fn ndims(&self) -> usize;
 
-    /// Embed multiple text documents in a single request
+    /// Embed multiple images in a single request from bytes.
     fn embed_images(
         &self,
         images: impl IntoIterator<Item = Vec<u8>> + Send,
     ) -> impl std::future::Future<Output = Result<Vec<Embedding>, EmbeddingError>> + Send;
 
-    /// Embed a single text document.
+    /// Embed a single image from bytes.
     fn embed_image<'a>(
         &'a self,
         bytes: &'a [u8],