display_name: Generates new tokens with a pre-trained model
type: command
compute: nas-gpu-cluster-NC6
inputs:
  pre_trained_model_path:
    type: uri_folder
    path: azureml://full/path/to/pretrained/model
  hub_tokenizer_path: gpt2
  prompt: Machine Learning
outputs:
  output_path:
    type: uri_file
code: .
environment:
  azureml:aml-archai:0.0.1
command: >-
  python generate_text.py
  ${{inputs.pre_trained_model_path}}
  ${{inputs.hub_tokenizer_path}}
  "${{inputs.prompt}}"
  --output_path ${{outputs.output_path}}