Put it to eval mode and sampeled some images

Add the checkpoint and the generated image

Dumb fix

Add the modelConfig to main

Add the trainer code

Finished the Sampler

Add the backbone and attention mechanism writing the sampler

initial commit