Path |
Digest |
Size |
flaxdiff/__init__.py |
sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU
|
0 |
flaxdiff/utils.py |
sha256=B0GcHlzlVYDNEIdh2v5qmP4u0neIT-FqexNohuyuCvg
|
2452 |
flaxdiff/data/__init__.py |
sha256=PM3PkHihyohT5SHVYKc8vQ4IeVfGPpCktkSVwvqMjQ4
|
52 |
flaxdiff/data/online_loader.py |
sha256=DoHrMZCi5gMd9tmkCpZIUU9lGxvfYtuaz58943_lCRc
|
11315 |
flaxdiff/models/__init__.py |
sha256=FAivVYXxM2JrCFIXf-C3374RB2Hth25dBrzOeNFhH1U
|
26 |
flaxdiff/models/attention.py |
sha256=JvrP7-09MV6IfRLRBhqjPmNUU-lkEMk9TOnJSBKcar8
|
13289 |
flaxdiff/models/common.py |
sha256=hWsSs2BP2J-JN1s4qLRr-h-KYkcVyl2hOp1Wsm_L-h8
|
10994 |
flaxdiff/models/favor_fastattn.py |
sha256=79Ew1nqarsNLPzZaBSd1ILORzJr74CupYeqGiCQK5E4
|
27689 |
flaxdiff/models/simple_unet.py |
sha256=L5m2j5580QP7pJ5VIme7U5xYA22PZiGP7qdvcKUnB38
|
11463 |
flaxdiff/models/simple_vit.py |
sha256=UCDDr0XVnpf6tbJWKFtEt3_nAqMqOoakXf5amyVWZNo
|
7929 |
flaxdiff/models/autoencoder/__init__.py |
sha256=qY-7MldZpsfkF-_T2LqlRK7VHbqfmosz0NmvzDlBkOk
|
78 |
flaxdiff/models/autoencoder/autoencoder.py |
sha256=27_hYl0yXAdH9Mx4Xu9J79mSNo-FEKr9SxhVaS3ffn4
|
591 |
flaxdiff/models/autoencoder/diffusers.py |
sha256=JHeFLCxiHhu-QHwhKiCuKsQJn4AZumquiuxgZkiYGQ0
|
3643 |
flaxdiff/models/autoencoder/simple_autoenc.py |
sha256=UXHPgDmwGTnv3Uts6Zj3p9R9nJXnEiEXbllgarwDfXM
|
805 |
flaxdiff/predictors/__init__.py |
sha256=SKkYYRF9Wfgk2zhtZw4vCXOdOeRlrm2Mk6cvuaEvAzc
|
4403 |
flaxdiff/samplers/__init__.py |
sha256=_S-9TwDeshrI0VmapV-J2hqjTByOa0-oOeUs_IdovjU
|
285 |
flaxdiff/samplers/common.py |
sha256=_an5h5Niz9Joz_-ppridLrGHpu8X0VVvhNGknPu6AUY
|
5272 |
flaxdiff/samplers/ddim.py |
sha256=XHMBX06S5hMTnKMaGh6fmq189pQcaGkA6fnX6YPbHP0
|
511 |
flaxdiff/samplers/ddpm.py |
sha256=d_58hfVShJHsRPQf5h1-4YKrD43-HGjWz9Vd8hltZBg
|
2627 |
flaxdiff/samplers/euler.py |
sha256=Epf7LBKUky7B8b-1ZyIlLWdRMgjmP08BQraGSKmr_3I
|
2726 |
flaxdiff/samplers/heun_sampler.py |
sha256=hhWnSM26OfOIFAcsuWYa1z-2QPjASuoYTop2byLWqzE
|
1388 |
flaxdiff/samplers/multistep_dpm.py |
sha256=ocmEq2sCvsULy6oTFaD5BhTU4c8VHsge4bdg6tfxW80
|
2724 |
flaxdiff/samplers/rk4_sampler.py |
sha256=BF-dMV1KauO-SYShqrCfm3U3V-1n4clqQXBeoG8RWQo
|
1728 |
flaxdiff/schedulers/__init__.py |
sha256=3id390WEfdf-MN-oLSPAhlRFIXrFWr6ioAHPAwURJyE
|
375 |
flaxdiff/schedulers/common.py |
sha256=b-W4iI-aqScpVE8VZbBpiYvAVI6rqDkUP-C_hEVBwCI
|
4151 |
flaxdiff/schedulers/continuous.py |
sha256=5c_niOA20fxJ5oJDi09FfayIRogBGwtfG0XThW2IUZk
|
334 |
flaxdiff/schedulers/cosine.py |
sha256=9ban0dFHLMm35wQvaBT4nCQwPGmzNsXwQ1xI0oppmJI
|
2005 |
flaxdiff/schedulers/discrete.py |
sha256=O54wH2HVu3olJA71NxgAXFW9cr6B6Gl-DR_uZeytpds
|
3319 |
flaxdiff/schedulers/exp.py |
sha256=cPTnUJpYdzJRRZqMLYQz0rRUCpEmaP2tXhRumLx94jA
|
605 |
flaxdiff/schedulers/karras.py |
sha256=4GN120kGwdxxU-h2mVdhBVy9IORkUMm_vvz3XjthBcI
|
3355 |
flaxdiff/schedulers/linear.py |
sha256=6003F5ISq1Wc0h6UAzY95MJgsDIKGMhBzbiVALpea0k
|
581 |
flaxdiff/schedulers/sqrt.py |
sha256=1F84ZgQPuoNMhe6yxGTR2G0h7dPOZtm4UDQOakbSsEU
|
445 |
flaxdiff/trainer/__init__.py |
sha256=T-vUVq4zHcMK6kpCsG4Gu8vn71q6lZD-lg-Ul7yKfEk
|
128 |
flaxdiff/trainer/autoencoder_trainer.py |
sha256=al7AsZ7yeDMEiDD-gbcXf0ADq_xfk1VMxvg24GfA-XQ
|
7008 |
flaxdiff/trainer/diffusion_trainer.py |
sha256=wKkg63DWZjx2MoM3VQNCDIr40rWN8fUGxH9jWWxfZao
|
9373 |
flaxdiff/trainer/simple_trainer.py |
sha256=cawm6fZNQoLLATMneAU2gQ9j7kefqHnBPHuaIj3i_a4
|
18237 |
flaxdiff-0.1.35.6.dist-info/METADATA |
sha256=NVCk5V7Zc3iq-nrWTivzO17dQa1fIjYgjJb800ZrZhQ
|
22085 |
flaxdiff-0.1.35.6.dist-info/WHEEL |
sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ
|
91 |
flaxdiff-0.1.35.6.dist-info/top_level.txt |
sha256=-2-nXnfkJgSfkki1tjm5Faw6Dso7vhtdn2szwCdX5CQ
|
9 |
flaxdiff-0.1.35.6.dist-info/RECORD |
— |
— |