easydel.utils.readme_generator

easydel.utils.readme_generator#

class easydel.utils.readme_generator.ModelInfo(name: str, type: str, repo_id: str, description: Optional[str] = None, model_type: Optional[str] = None, model_task: str = 'CausalLM', attn_mechanism: str = 'vanilla', features: Optional[List[str]] = None, supported_tasks: Optional[List[str]] = None, limitations: Optional[List[str]] = None, version: str = '0.1.4', overview: Optional[str] = None)[source]#

Bases: object

Model information container. Used to pass data to the Jinja template.

attn_mechanism: str = 'vanilla'#
description: Optional[str] = None#
features: Optional[List[str]] = None#
limitations: Optional[List[str]] = None#
model_task: str = 'CausalLM'#
model_type: Optional[str] = None#
name: str#
overview: Optional[str] = None#
repo_id: str#
supported_tasks: Optional[List[str]] = None#
type: str#
version: str = '0.1.4'#
class easydel.utils.readme_generator.ReadmeGenerator(template_dir: Optional[str] = None)[source]#

Bases: object

Generate README files for EasyDeL models.

generate_readme(model_info: ModelInfo, output_path: Optional[str] = None) → str[source]#

Generate README content for a model.

Parameters

  • model_info – Model information

  • output_path – tp.Optional path to save the README

Returns

Generated README content

Contents