Commit Graph

24 Commits

Author SHA1 Message Date
lopho 9916294de1
fix wandb init mode, don't log hf token
correct value for mode ('enabled' is invalid)
clear hf_token passed to wandb to avoid logging it
2022-11-16 22:28:16 +01:00
Anthony Mercurio dc5849b235
Merge branch 'main' into inference-option 2022-11-16 16:20:57 -05:00
chavinlo a2772fc668 fixes 2022-11-16 10:55:38 -05:00
chavinlo fed3431f03 Revert "sync trainer with main branch"
This reverts commit 80e2422967.
2022-11-16 10:44:39 -05:00
Carlos Chavez 80e2422967
sync trainer with main branch 2022-11-16 10:39:20 -05:00
Maw-Fox 015eeae274
Documentation consistency. 2022-11-15 10:34:55 -07:00
Anthony Mercurio 29ffbd645e
Fix noise scheduler 2022-11-15 11:08:38 -05:00
Maw-Fox 6c5b2e7149
Fix of fix 2022-11-15 07:15:18 -07:00
Maw-Fox 2c18d29613
Fix from upstream merge. 2022-11-15 06:42:14 -07:00
Carlos Chavez d600078008
Add options and local inference
Added options to:
- Disable Inference (it consumes about 2gb of VRAM even when not active)
- Disable wandb

and:
- if no hftoken is provided it just fills it with nothing so it doesn't argues
- if wandb is not enabled, save the inference outputs to a local folder along with information about it
2022-11-14 22:08:16 -05:00
Maw-Fox 773e65f324
Merge origin:main into remote:staging-migration 2022-11-14 19:59:45 -07:00
Maw-Fox 95b9407a3e
Add+config .gitignore (bring back git stage) and fix up documentation. 2022-11-12 18:48:16 -07:00
Maw-Fox 6bd6c6a4ef
Fixed/flipped help text. 2022-11-12 16:11:30 -07:00
Maw-Fox 189f621a1e
Here, let's fix this while we're at it. 2022-11-12 15:47:17 -07:00
lopho cd0910e82d
Parse booleans in argument parser
true, yes or 1 correspond to True, else False.
2022-11-12 11:14:48 +01:00
Maw-Fox d1eb3ace3f I lied. 2022-11-11 18:17:29 -07:00
Maw-Fox 6c2d5d8066 Final cleanup. 2022-11-11 18:09:09 -07:00
Maw-Fox de221ea42e Derp. ImageStore.__init__ already iterates fully :) 2022-11-11 17:59:32 -07:00
Maw-Fox 925eacf374 Cleanup 2022-11-11 17:50:23 -07:00
Maw-Fox c12cbfced3 Fixed ref typo. 2022-11-11 17:43:09 -07:00
Maw-Fox 6480336d2c Cleanup test code. 2022-11-11 17:17:50 -07:00
Maw-Fox 120d406355 Implementation of validation/resize classes. 2022-11-11 17:14:46 -07:00
Anthony Mercurio 624f0f14af
correctly set args 2022-11-11 08:25:27 -07:00
harubaru 1f5b671b67 relicense 2022-11-10 12:59:53 -07:00