"third_party/HugeCTR/vscode:/vscode.git/clone" did not exist on "3befaca205bd387a74d7720712286346847e0bb8"
  • Simon Kirsten's avatar
    Flax: Fix img2img and align with other pipeline (#1824) · ab0e92fd
    Simon Kirsten authored
    
    
    * Flax: Add components function
    
    * Flax: Fix img2img and align with other pipeline
    
    * Flax: Fix PRNGKey type
    
    * Refactor strength to start_timestep
    
    * Fix preprocess images
    
    * Fix processed_images dimen
    
    * latents.shape -> latents_shape
    
    * Fix typo
    
    * Remove "static" comment
    
    * Remove unnecessary optional types in _generate
    
    * Apply doc-builder code style.
    Co-authored-by: default avatarPedro Cuenca <pedro@huggingface.co>
    ab0e92fd
unet_2d_condition_flax.py 13.7 KB