Skip to content

Commit

Permalink
fix long names
Browse files Browse the repository at this point in the history
  • Loading branch information
zachcp committed Nov 15, 2024
1 parent 8eae938 commit 8512042
Show file tree
Hide file tree
Showing 4 changed files with 17 additions and 3 deletions.
6 changes: 6 additions & 0 deletions candle-transformers/src/models/bert.rs
Original file line number Diff line number Diff line change
@@ -1,3 +1,9 @@
//! BERT (Bidirectional Encoder Representations from Transformers)
//!
//! See "BERT: Pre-training of Deep Bidirectional Transformers for Language Understanding", Devlin et al. 2018
//! - [Arxiv](https://arxiv.org/abs/1810.04805)
//! - [Github](https://github.com/google-research/bert)
//!
use super::with_tracing::{layer_norm, linear, LayerNorm, Linear};
use candle::{DType, Device, Result, Tensor};
use candle_nn::{embedding, Embedding, Module, VarBuilder};
Expand Down
2 changes: 1 addition & 1 deletion candle-transformers/src/models/moondream.rs
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
//! MoonDream Model provides vision-to-text capabilities based on Microsoft's Phi architecture
//! MoonDream Model vision-to-text
//!
//! The model consists of:
//! - Vision encoder using a ViT-style architecture
Expand Down
4 changes: 2 additions & 2 deletions candle-transformers/src/models/persimmon.rs
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
//! Persimmon Model, a transformer language model for efficient inference and general-purpose tasks
//! Persimmon Model
//!
//! See Persimmon model details at:
//! A transformer language model for efficient inference and general-purpose tasks. See Persimmon model details at:
//! - [Hugging Face](https://huggingface.co/adept/persimmon-8b-base)
//!
//! The model uses a standard transformer architecture with:
Expand Down
8 changes: 8 additions & 0 deletions candle-transformers/src/models/segment_anything/mod.rs
Original file line number Diff line number Diff line change
@@ -1,3 +1,11 @@
//! Segment Anything Model (SAM)
//!
//! SAM is an architecture for image segmentation, capable of segmenting any object
//! in an image based on prompts like points or boxes.
//!
//! - [GH Link](https://github.com/facebookresearch/segment-anything)
//! - [Paper](https://arxiv.org/abs/2304.02643)
//!
pub use crate::models::with_tracing::Linear;
use candle::{Result, Tensor};
use candle_nn::{Module, VarBuilder};
Expand Down

0 comments on commit 8512042

Please sign in to comment.