Skip to main content

Model.download

Model.download(name=None, location=None, framework=None)

     This method downloads the model to a specified location on your local machine. By default,
     it will download the model to the current working directory in the Ludwig format.
     to the current working directory in the Ludwig format.

Parameters:​

       name: Optional[str]
       Name of the compressed model folder to download.

       location: Optional[str]
       File path to download the model to. By default, this method will download to the current working
       directory. The specified path can be a relative or an absolute path.

       framework: Optional[str]
       Framework to download the model as. Default is to download as a Ludwig Model. Available frameworks
       are: 'ludwig', 'torchscript', and 'triton'.

Returns:​

       None

Examples:​

Get a model and download it.

    model = pc.get_model(model_id=5678)

    # Download to current working directory
    model.download()

    # Download 'ludwig_model' to a specific location and with 'triton' framework
    model.download(name='ludwig_model.zip', location='/Users/jdoe/Downloads', framework='triton')