Skip to content

Commit

Permalink
doc update
Browse files Browse the repository at this point in the history
  • Loading branch information
zachcp committed Dec 29, 2024
1 parent 7e463fa commit ab7dc09
Show file tree
Hide file tree
Showing 2 changed files with 0 additions and 20 deletions.
12 changes: 0 additions & 12 deletions ferritin-onnx-models/src/models/esm2/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -8,18 +8,6 @@
//! * ESM2_T12_35M - medium 12-layer protein language model
//! * ESM2_T30_150M - large 30-layer protein language model
//!
//! # Example
//! ```no_run
//! use anyhow::Result;
//! use ferritin_onnx_models::ESM2;
//!
//! let tokenizer = ESM2::load_tokenizer()?;
//! let text = "MLKLRV";
//! let encoding = tokenizer.encode(text, false)?;
//! let tokens = encoding.get_tokens();
//! assert_eq!(tokens.len(), 6);
//! assert_eq!(tokens, &["M", "L", "K", "L", "R", "V"]);
//! ```
use anyhow::{anyhow, Result};
use candle_hf_hub::api::sync::Api;
Expand Down
8 changes: 0 additions & 8 deletions ferritin-onnx-models/src/models/ligandmpnn/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -5,14 +5,6 @@
//!
//! The models are loaded from the Hugging Face model hub and executed using ONNX Runtime.
//!
//! # Examples
//! ```no_run
//! use ferritin_core::AtomCollection;
//! use ferritin_onnx_models::LigandMPNN;
//! let model = LigandMPNN::new().unwrap();
//! let atom_collection = AtomCollection::new();
//! let logits = model.run_model(atom_collection, 10, 0.1).unwrap();
//! ```
//!
use crate::{ndarray_to_tensor_f32, tensor_to_ndarray_f32, tensor_to_ndarray_i64};
use anyhow::Result;
Expand Down

0 comments on commit ab7dc09

Please sign in to comment.