Feature Suggestions

[benearnthof]

Add feature suggestions below:

Current Goals:

• [x] Huber Loss
• [x] LPIPS Loss
• [ ] Modularize Code
• [ ] Add Sampling To Model
• [x] Add Unet & Required Modules
• [ ] Add conditioning wrapper to Unet for image to image tasks
• [ ] Add Trainer with Accelerate
• [ ] Add Readme
• [ ] Add Model Classes for Latent Flow
• [ ] Add Rectified Flow & Other potential Flow Matching variants (please suggest papers below)
• [ ] Flow Map Matching seems very interesting https://arxiv.org/abs/2406.07507

[benearnthof]

added huber and lpips loss from rectified flow pytorch will modularize code now to make everything more flexible

[benearnthof]

Great Theoretical Paper: https://arxiv.org/pdf/2402.03232