( self, vocab_size: int, num_layers: int, dims: int, num_heads: int,

( self, num_layers: int, input_dim: int, hidden_dim: int, output_dim: int )

( self, input_dims: int, output_dims: int, lora_rank: int = 8, bias: b

(self, args: ModelArgs)

(self, args: ModelArgs)

(self, args: ModelArgs)

(self, args: ModelArgs)

(self, path: Path, key: str = "text")

(self, dataset, save_dir="/tmp")

(self, shift_and_log_scale: mx.array)

(self, mu: mx.array, sigma: mx.array)

(self, n_transforms: int, d_params: int, d_hidden: int, n_layers: int)

(self, bpe_ranks, vocab)

( self, crop_size: int = 224, do_center_crop: bool = True, do_normalize: bool

( self, dims: int, num_heads: int, query_input_dims: Optional[int] = None,

(self, config: CLIPTextConfig)

(self, config: CLIPTextConfig)

(self, config: CLIPTextConfig)

(self, config: CLIPTextConfig)

(self, config: CLIPVisionConfig)

(self, config: CLIPVisionConfig)

(self, config: CLIPConfig)

(self, dim, hidden_dim, dropout=0.0)

(self, dim, heads, dropout=0.0)

(self, dim, heads, mlp_dim, dropout=0.0)

(self, dim, depth, heads, mlp_dim, dropout=0.0)

( self, dims: int, num_heads: int, mlp_dims: Optional[int] = None, lay

( self, num_layers: int, dims: int, num_heads: int, mlp_dims: Optional[int] = None )

(self, config)

(self, config: LlaVAConfig)

(self, config: TextConfig)

(self, config: TextConfig)

(self, config: TextConfig)

(self, config: TextConfig)

( self, dims: int, num_heads: int, query_input_dims: Optional[int] = None,

(self, config: VisionConfig)

(self, config: VisionConfig)

(self, config: VisionConfig)

(self, config: VisionConfig)

(self, config: VisionConfig)

(self, dims)

(self, block, num_blocks, num_classes=10)

(self, args: ModelArgs)

(self, args: ModelArgs)

(self, args: ModelArgs)

(self, args: ModelArgs)

(self, model_path: str)

(self, args: ModelArgs)

(self, args: ModelArgs)

(self, args: ModelArgs)

(self, args: ModelArgs)

(self, metadata)

(self, args: ModelArgs)

(self, args: ModelArgs)

(self, args: ModelArgs)

(self, args: ModelArgs)

(self, args: ModelArgs)

(self, args: ModelArgs)

(self, model_path: str)

(self, model_name: str)

( self, model: Model, draft_model: Model, tokenizer: str, num_draft: i

(self, config: T5Config, bidirectional: bool)

(self, config: T5Config)

(self, config: T5Config)

(self, config: T5Config)

(self, config: T5Config)

(self, config: T5Config)

(self, config: T5Config)

(self, config: T5Config)

(self, model: "Whisper")

(self, length_penalty: Optional[float])

(self, temperature: float, eot: int)

(self, tokenizer: Tokenizer, sample_begin: int, n_vocab: int)

(self, suppress_tokens: Sequence[int], n_vocab: int)

( self, tokenizer: Tokenizer, sample_begin: int, max_initial_timestamp_index:

(self, model: "Whisper", options: DecodingOptions)

(self, output_dir: str)

(self, n_state: int, n_head: int)

(self, n_state: int, n_head: int, cross_attention: bool = False)

( self, n_mels: int, n_ctx: int, n_state: int, n_head: int, n_layer: int )

( self, n_vocab: int, n_ctx: int, n_state: int, n_head: int, n_layer: int )

(self, n_state: int, n_head: int)

(self, n_state: int, n_head: int, cross_attention: bool = False)

( self, n_mels: int, n_ctx: int, n_state: int, n_head: int, n_

( self, n_vocab: int, n_ctx: int, n_state: int, n_head: int, n

( self, config, in_channels: int, out_channels: int, kernel_size: int,

( self, config, in_channels: int, out_channels: int, kernel_size: int,

(self, config, dimension)

(self, config, dim: int, dilations: List[int])

(self, config)

(self, config)

(self, config)

(self, config)

(self, config)

(self, config)

(self, in_channels: int, time_embed_dim: int)

( self, model_dims: int, num_heads: int, hidden_dims: Optional[int] = None,

( self, in_channels: int, model_dims: int, encoder_dims: int, num_head

( self, in_channels: int, out_channels: Optional[int] = None, groups: int = 32