experimental_experiment.torch_models

modules

• experimental_experiment.torch_models.diffusion_model_helper
• experimental_experiment.torch_models.dump_helper
• experimental_experiment.torch_models.llama_helper
• experimental_experiment.torch_models.llm_model_helper
• experimental_experiment.torch_models.mistral_helper
• experimental_experiment.torch_models.phi3_helper
• experimental_experiment.torch_models.phi_helper
• experimental_experiment.torch_models.training_helper

experimental_experiment.torch_models.assert_found(kwargs: Dict[str, Any], config: Dict[str, Any])

Checks a parameter is available.

experimental_experiment.torch_models.flatten_outputs(output: Any) → Iterator[Any]

Flattens output results.