• oahzxl's avatar
    [autochunk] support diffusion for autochunk (#2621) · 6ba83648
    oahzxl authored
    * add alphafold benchmark
    
    * renae alphafold test
    
    * rename tests
    
    * rename diffuser
    
    * renme
    
    * rename
    
    * update transformer
    
    * update benchmark
    
    * update benchmark
    
    * update bench memory
    
    * update transformer benchmark
    
    * rename
    
    * support diffuser
    
    * support unet metainfo prop
    
    * fix bug and simplify code
    
    * update linear and support some op
    
    * optimize max region search, support conv
    
    * update unet test
    
    * support some op
    
    * support groupnorm and interpolate
    
    * update flow search
    
    * add fix dim in node flow
    
    * fix utils
    
    * rename
    
    * support diffusion
    
    * update diffuser
    
    * update chunk search
    
    * optimize imports
    
    * import
    
    * finish autochunk
    6ba83648
search_chunk.py 12.7 KB