Add the checkpoint and the generated image

Add the backbone and attention mechanism writing the sampler