updated to the newest version of burn

This commit is contained in:
rasmus
2026-03-03 15:12:59 +01:00
committed by Ben_Kosytorz
parent 75f0cedd9f
commit 6a73e7b27e
8 changed files with 39 additions and 43 deletions

View File

@@ -15,7 +15,7 @@ use burn::{
//use crate::backend::Backend as MyBackend;
use crate::backend::{qkv_attention, attn_decoder_mask};
#[derive(Config)]
#[derive(Config, Debug)]
pub struct CLIPConfig {
n_vocab: usize,
n_state: usize,
@@ -75,7 +75,7 @@ impl<B: Backend> CLIP<B> {
}
}
#[derive(Config)]
#[derive(Config, Debug)]
pub struct ResidualDecoderAttentionBlockConfig {
n_state: usize,
n_head: usize,
@@ -114,7 +114,7 @@ impl<B: Backend> ResidualDecoderAttentionBlock<B> {
}
}
#[derive(Config)]
#[derive(Config, Debug)]
pub struct MultiHeadSelfAttentionConfig {
n_state: usize,
n_head: usize,