INNER CODE UNIT · TypeScript
ConfigGenerator
theaniketgiri/create-llm · src/config-generator.ts:7
export class ConfigGenerator {
/**
* Generate llm.config.js content based on template and user config
*/
generateConfig(projectConfig: ProjectConfig, template: Template): string {
const { model, training, data, tokenizer } = template.config;
const config = {
model: {
type: model.type,
size: projectConfig.template,
vocab_size: model.vocab_size,
max_length: model.max_length,
layers: model.layers,
heads: model.heads,
dim: model.dim,
dropout: model.dropout
},