Updated doctests

src/distilbert/mod.rs

@@ -44,7 +44,7 @@
 //! let weights_path = download_resource(&weights_resource)?;
 //! let device = Device::cuda_if_available();
 //! let mut vs = nn::VarStore::new(device);
-//! let tokenizer: BertTokenizer = BertTokenizer::from_file(vocab_path.to_str().unwrap(), true);
+//! let tokenizer: BertTokenizer = BertTokenizer::from_file(vocab_path.to_str().unwrap(), true)?;
 //! let config = DistilBertConfig::from_file(config_path);
 //! let bert_model = DistilBertModelMaskedLM::new(&vs.root(), &config);
 //! vs.load(weights_path)?;

src/electra/mod.rs

@@ -46,7 +46,7 @@
 //! let weights_path = download_resource(&weights_resource)?;
 //! let device = Device::cuda_if_available();
 //! let mut vs = nn::VarStore::new(device);
-//! let tokenizer: BertTokenizer = BertTokenizer::from_file(vocab_path.to_str().unwrap(), true);
+//! let tokenizer: BertTokenizer = BertTokenizer::from_file(vocab_path.to_str().unwrap(), true)?;
 //! let config = ElectraConfig::from_file(config_path);
 //! let electra_model = ElectraForMaskedLM::new(&vs.root(), &config);
 //! vs.load(weights_path)?;

src/gpt2/mod.rs

@@ -46,7 +46,7 @@
 //!     vocab_path.to_str().unwrap(),
 //!     merges_path.to_str().unwrap(),
 //!     true,
-//! );
+//! )?;
 //! let config = Gpt2Config::from_file(config_path);
 //! let gpt2_model = GPT2LMHeadModel::new(&vs.root(), &config);
 //! vs.load(weights_path)?;

src/openai_gpt/mod.rs

@@ -46,7 +46,7 @@
 //!     vocab_path.to_str().unwrap(),
 //!     merges_path.to_str().unwrap(),
 //!     true,
-//! );
+//! )?;
 //! let config = Gpt2Config::from_file(config_path);
 //! let gpt_model = OpenAiGptModel::new(&vs.root(), &config);
 //! vs.load(weights_path)?;

src/roberta/mod.rs

@@ -52,7 +52,7 @@
 //!     vocab_path.to_str().unwrap(),
 //!     merges_path.to_str().unwrap(),
 //!     true,
-//! );
+//! )?;
 //! let config = BertConfig::from_file(config_path);
 //! let bert_model = RobertaForMaskedLM::new(&vs.root(), &config);
 //! vs.load(weights_path)?;