Add the checkpoint and the generated image

Finished the Sampler

Add the backbone and attention mechanism writing the sampler