Surly Marike. Jr issues

Repositories
Issues
Comments

Results 2 issues of


                                            Surly Marike. Jr

[OOM] Does the maker know what those 'partial(jax.pmap...' actually do?

Or just mindlessly copypasta? I tried to solve the OOM problem and found out that those replicate(...) /flax.jax_utils.replicate are the major cause. That replicate thing literally doubled the ram usage...

Train SR model with Text-to-Image latents

I noticed the SR / Super-resolution model takes `[1,3,H,W]` input (original pixels?). And before-first_stage_model txt2img output is `[1,4,H>>3,W>>3]`. Is it possible to get the SR model works with `[1,4,H>>3,W>>3]` latent?...