• Isamu Isozaki's avatar
    Retrieval Augmented Diffusion Models (#3297) · 3201903d
    Isamu Isozaki authored
    
    
    * Resetting rdm pr
    
    * Fixed styles
    
    * Fixed style
    
    * Moved to rdm folder+fixed slight errors
    
    * Removed config diff
    
    * Started adding tests
    
    * Adding retrieved images
    
    * Fixed faiss import
    
    * Fixed import errors
    
    * Fixing tests
    
    * Added require_faiss
    
    * Updated dependency table
    
    * Attempt solving consistency test
    
    * Fixed truncation and vocab size issue
    
    * Passed common tests
    
    * Finished up cpu testing on pipeline
    
    * Passed all tests locally
    
    * Removed some slow tests
    
    * Removed diffs from test_pipeline_common
    
    * Remove logs
    
    * Removed diffs from test_pipelines_common
    
    * Fixed style
    
    * Fully fixed styles on diffs
    
    * Fixed name
    
    * Proper rename
    
    * Fixed dummies
    
    * Fixed issue with dummyonnx
    
    * Fixed black style
    
    * Fixed dummies
    
    * Changed ordering
    
    * Fixed logging
    
    * Fixing
    
    * Fixing
    
    * quality
    
    * Debugging regex
    
    * Fix dummies with guess
    
    * Fixed typo
    
    * Attempt fix dummies
    
    * black
    
    * ruff
    
    * fixed ordering
    
    * Logging
    
    * Attempt fix
    
    * Attempt fix dummy
    
    * Attempt fixing styles
    
    * Fixed faiss dependency
    
    * Removed unnecessary deprecations
    
    * Finished up main changes
    
    * Added doc
    
    * Passed tests
    
    * Fixed tests
    
    * Remove invisible watermark
    
    * Fixed ruff errors
    
    * Added prompt embed to tests
    
    * Added tests and made retriever an optional component
    
    * Fixed styles
    
    * Made faiss a dependency of pipeline
    
    * Logging
    
    * Fixed dummies
    
    * Make pipeline test work
    
    * Fixed style
    
    * Moved to research projects
    
    * Remove diff
    
    * Fixed style error
    
    ---------
    Co-authored-by: default avatarPatrick von Platen <patrick.v.platen@gmail.com>
    3201903d
retriever.py 8.81 KB