INNER CODE UNIT · TypeScript

params

theaniketgiri/create-llm · src/prompts.ts:58

      const params = (template.config.model.parameters / 1_000_000).toFixed(0);
      const hardware = template.config.hardware.can_run_on_cpu ? 'CPU' : template.config.hardware.recommended_gpu;
      const time = template.config.hardware.estimated_training_time;
      
      return {
        name: `${chalk.bold(template.name.toUpperCase())} - ${template.config.documentation.description}\n  ${chalk.gray(`${params}M params | ${hardware} | ${time}`)}`,
        value: template.name,
        short: template.name
      };
    });

    const { template } = await inquirer.prompt([
      {
        type: 'list',
        name: 'template',
        message: 'Select a template:',
        choices,
        default: defaultTemplate || 'small',

View source record →