diff options
| author | Juarez Bochi <juarez.bochi@grammarly.com> | 2023-11-06 23:35:37 -0500 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-11-07 05:35:37 +0100 |
| commit | 508f811b93035f076e18778fe08106f15abfa8a7 (patch) | |
| tree | b31cf3c6bbaf335ab8371f71a1353fe597bab8fd /candle-transformers/src/quantized_var_builder.rs | |
| parent | a773a4b22b88d9955f51de552d72717441d49729 (diff) | |
| download | candle-508f811b93035f076e18778fe08106f15abfa8a7.tar.gz candle-508f811b93035f076e18778fe08106f15abfa8a7.tar.bz2 candle-508f811b93035f076e18778fe08106f15abfa8a7.zip | |
Add support for MADLAD400 (#1285)
* Add support for madlad
* Add support for quantized MADLAD
Diffstat (limited to 'candle-transformers/src/quantized_var_builder.rs')
| -rw-r--r-- | candle-transformers/src/quantized_var_builder.rs | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/candle-transformers/src/quantized_var_builder.rs b/candle-transformers/src/quantized_var_builder.rs index 810802e8..63101f4c 100644 --- a/candle-transformers/src/quantized_var_builder.rs +++ b/candle-transformers/src/quantized_var_builder.rs @@ -90,4 +90,8 @@ impl VarBuilder { pub fn device(&self) -> &Device { &self.device } + + pub fn contains_key(&self, key: &str) -> bool { + self.data.contains_key(key) + } } |