flaxmodels

flaxmodels copied to clipboard

It doesn’t look like there’s a way to remove the head in the resnet models. There is a way to dump the activations. It would be nice if I could...

vyeevani

Use `safetensors` to store tensors instead of `pickle`

3

Hi @matthias-wright, I've been playing around for a couple days with your project and it's so cool, thanks for building some pure `flax` models here 👍🏻 Don't know if you're...

alvarobartt

Possible bug in StyleGAN training code

1

Hi, Nice repo! Is this line a bug? Since I think `batch['images']` is `N x B x H x W x C`, so the indices should be shift up by...

wilson1yan

``` Downloading: "https://www.dropbox.com/s/e8de1peq7p8gq9d/stylegan2_generator_ffhq.h5" to /tmp/flaxmodels/stylegan2_generator_ffhq.h5 100%|██████████| 133M/133M [00:02 15 params = generator.init(key, z) 16 images = generator.apply(params, z, train=False) 17 [... skipping hidden 9 frame] 8 frames [/usr/local/lib/python3.12/dist-packages/flaxmodels/stylegan2/generator.py](https://localhost:8080/#) in __call__(self,...

Skylion007