huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 19:18:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix bug introduced in madlad PR (#1298)

Browse Source
This commit is contained in:
Juarez Bochi
2023-11-08 11:55:46 -05:00
committed by GitHub
parent 2feb0b054f
commit f772213e84
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
candle-transformers/src/models/quantized_t5.rs
View File

@ -644,7 +644,7 @@ pub struct T5EncoderModel {
impl T5EncoderModel { impl T5EncoderModel {
pub fn load(vb: VarBuilder, cfg: &Config) -> Result<Self> { pub fn load(vb: VarBuilder, cfg: &Config) -> Result<Self> {
let shared_vb = if vb.contains_key("shared") { let shared_vb = if vb.contains_key("shared.weight") {
vb.pp("shared") vb.pp("shared")
} else { } else {
vb.pp("decoder").pp("embed_tokens") vb.pp("decoder").pp("embed_tokens")
@ -690,7 +690,7 @@ impl T5ForConditionalGeneration {
pub fn load(vb: VarBuilder, cfg: &Config) -> Result<Self> { pub fn load(vb: VarBuilder, cfg: &Config) -> Result<Self> {
assert!(cfg.is_encoder_decoder); assert!(cfg.is_encoder_decoder);
let d_model = cfg.d_model; let d_model = cfg.d_model;
let shared_vb = if vb.contains_key("shared") { let shared_vb = if vb.contains_key("shared.weight") {
vb.pp("shared") vb.pp("shared")
} else { } else {
vb.pp("decoder").pp("embed_tokens") vb.pp("decoder").pp("embed_tokens")

4
candle-transformers/src/models/t5.rs
View File

@ -670,7 +670,7 @@ pub struct T5EncoderModel {
impl T5EncoderModel { impl T5EncoderModel {
pub fn load(vb: VarBuilder, cfg: &Config) -> Result<Self> { pub fn load(vb: VarBuilder, cfg: &Config) -> Result<Self> {
let shared_vb = if vb.contains_tensor("shared") { let shared_vb = if vb.contains_tensor("shared.weight") {
vb.pp("shared") vb.pp("shared")
} else { } else {
vb.pp("decoder").pp("embed_tokens") vb.pp("decoder").pp("embed_tokens")
@ -716,7 +716,7 @@ impl T5ForConditionalGeneration {
pub fn load(vb: VarBuilder, cfg: &Config) -> Result<Self> { pub fn load(vb: VarBuilder, cfg: &Config) -> Result<Self> {
assert!(cfg.is_encoder_decoder); assert!(cfg.is_encoder_decoder);
let d_model = cfg.d_model; let d_model = cfg.d_model;
let shared_vb = if vb.contains_tensor("shared") { let shared_vb = if vb.contains_tensor("shared.weight") {
vb.pp("shared") vb.pp("shared")
} else { } else {
vb.pp("decoder").pp("embed_tokens") vb.pp("decoder").pp("embed_tokens")