AI-DungeonMaster
AI-DungeonMaster copied to clipboard
Pass temperature and top_k to sampler in generator.py
In generator.py I changed:
_self.output = sample.sample_sequence(
hparams=hparams, length=length,
context=self.context,
batch_size=batch_size,
)_
to self.output = sample.sample_sequence( hparams=hparams, length=length, context=self.context, batch_size=batch_size, temperature=temperature, top_k=top_k )
I am trying some finetuned models and those values make a world of difference in coherence and quality.