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: Optional[str] = None, features: Optional[List[str]] = None, supported_tasks: Optional[List[str]] = None, limitations: Optional[List[str]] = None, version: str = '0.1.0')[source]#

Bases: object

Model information container.

description: Optional[str] = None#
features: Optional[List[str]] = None#
limitations: Optional[List[str]] = None#
model_task: Optional[str] = None#
model_type: Optional[str] = None#
name: str#
repo_id: str#
supported_tasks: Optional[List[str]] = None#
type: str#
version: str = '0.1.0'#
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

  • template_name – Name of the template to use

Returns

Generated README content

Contents