Functions — github.com/Standard-Intelligence/hertz-dev

hub / github.com/Standard-Intelligence/hertz-dev / functions

Functions173 in github.com/Standard-Intelligence/hertz-dev

⨍ Functions 173◇ Types & classes 41 ↳ Endpoints 1

( self, dim: int, end: int, theta: float = 10_000, )

transformer.py:50

Method__init__

(self, *args, **kwargs)

transformer.py:137

Method__init__

(self, dim: int, eps: float = 1e-5,)

transformer.py:141

Method__init__

(self, dim: int, expand_dim: int = None,)

transformer.py:153

Method__init__

(self, dim: int, n_head: int, shape_rotator: ShapeRotator, kv

transformer.py:236

Method__init__

(self, dim: int, ff_dim: int, eps: float = 1e-5,)

transformer.py:255

Method__init__

(self, dim: int, layer_id: int = 0, n_head: int = 16, kv_heads

transformer.py:267

Method__init__

(self, dim, vocab_size)

(self, dim, vocab_size)

(self, *args, **kwargs)

utils/dist.py:27

Method_apply_weight_norm

(m)

tokenizer.py:402

Method_remove_weight_norm

Method_reset_parameters

(m)

tokenizer.py:394

Methodaudio_callback

This is called for each audio block

inference_client.py:41

Methodcompletion

only accepts latent-space data.

Args: x (Tensor): Float tensor variable with the shape (B, C, T). Returns: Tensor: Float tensor variable wit

tokenizer.py:52

Methodforward

Args: x (Tensor): Float tensor variable with the shape (B, C, T). Returns: Tensor: Float tensor variable wit

(self, x, return_feats=False)

(self, input: Tensor)

x: (B, S, D) kv: (B, S, H, D)

transformer.py:224

Methodforward

x: (B, S, D) kv: (B, S, H, D)

x: (B, S, D) kv: (B, S, H, D)

einstein notation b - batch n - sequence (or flattened spatial dimensions) d - feature dimension c - number o

(self, data, next_tokens: Optional[Tuple[T.Tensor, T.Tensor]] = None, temps: Optional[Tuple[float, Tuple[float

model.py:139

Functionget_activation

(nonlinear_activation, nonlinear_activation_params={})

(self, bsize, device, dtype, length:int=None)

(self, codes, get_global=False)

ioblocks.py:203

Functionlayer_param_distribution

(module)

utils/interp.py:59

Methodloss

(self, data, dataHat)

ioblocks.py:55

Functionmodel_size_estimation

(module)

utils/interp.py:44

Methodneeds_float32_params

(self)

ioblocks.py:137

Functionnew_init

(self, *args, **kwargs)

utils/blocks.py:38

Methodoutput_stream_callback

Callback for output stream to get audio data

inference_client.py:52

Methodqueued_audio_frames_callback

(self, frames: List[av.AudioFrame])

inference_client_webrtc.py:118

Methodrepr_from_latent

Functionround_up_multiple

(num, mult)

ioblocks.py:106

Functionset_temperature

(token_temp: Optional[float] = None, categorical_temp: Optional[float] = None, gaussian_temp: Optional[float]

inference_server.py:132

Functionsi_module

(cls: Type[si_module_TpV])

utils/blocks.py:13

Methodstop

(self)

inference_client_webrtc.py:176

Functionwebsocket_endpoint

(websocket: WebSocket)

inference_server.py:144

← previous101–173 of 173, ranked by callers