Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Saiyan-World/goku
/ functions
Functions
16 in github.com/Saiyan-World/goku
⨍
Functions
16
◇
Types & classes
7
↓ 6 callers
Method
__init__
(self, dim, num_attention_heads, attention_head_dim, dropout=0.0, cross_attention_dim=None, attention_
goku/model.py:189
↓ 2 callers
Function
apply_rotary_emb
( x: torch.Tensor, freqs_cis: Union[torch.Tensor, Tuple[torch.Tensor]], )
goku/model.py:9
↓ 1 callers
Method
prepare_attention_mask
( # https://github.com/huggingface/diffusers/blob/main/src/diffusers/models/attention_processor.py#L69
goku/model.py:109
Method
__init__
(self, patch_size, in_channels, embed_dim, bias=True)
goku/model.py:26
Method
__init__
(self, embedding_dim)
goku/model.py:41
Method
__init__
(self, dim, dim_out=None, mult=4, inner_dim=None, bias=True)
goku/model.py:65
Method
__init__
(self, dim: int, eps=1e-6)
goku/model.py:80
Method
__init__
(self, q_dim, kv_dim=None, heads=8, dim_head=64, dropout=0.0, bias=False)
goku/model.py:91
Method
__init__
(self, in_channels, out_channels, num_attention_heads, attention_head_dim, depth, patch_size=2, dropou
goku/model.py:238
Method
forward
(self, x)
goku/model.py:30
Method
forward
(self, x, timestep, seq_len_list=None)
goku/model.py:48
Method
forward
(self, hidden_states)
goku/model.py:72
Method
forward
(self, x: torch.Tensor)
goku/model.py:85
Method
forward
( self, inputs_q, inputs_kv, attention_mask=None, cross_attention=Fals
goku/model.py:130
Method
forward
( self, hidden_states, encoder_hidden_states=None, encoder_attention_mask=None
goku/model.py:208
Method
forward
(self, hidden_states, encoder_hidden_states, timestep, rope_pos_embed, encoder_attention_mask)
goku/model.py:261