Skip to content

Conversation

balakrish181
Copy link

Summary:

  • Created the sampler.py script to allow users to generate images from pre-trained model weights.
  • This change directly addresses Issue How to generate images through trained model files? #353 where users asked how to use the trained model files (.pt) to generate images.
  • The new script makes it easy to generate images using saved model checkpoints (e.g., model-{epoch}.pt).

Changes:

  • Added the sampler.py script for image generation from trained models.
  • Provided clear argument handling to allow users to specify the number of images, batch size, image size, and checkpoint to load.
  • Updated markdown documentation to reflect usage instructions and examples, ensuring clarity for end-users.

Issue Resolved:

@balakrish181 balakrish181 changed the title Added sampler.py for Image Generation from Trained Model Weights (Addresses Issue #353) Image Sampling from Saved Model Weights (Addresses Issue #353) Apr 4, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant