Commit History
fix: move to cuda before generate
2fa3fb0
fix: map to cpu
a4e01fa
fix: no map location in main process
330e8a2
fix: move to cuda in generate function
c1cc7f4
feat: ZeroGPU implementation
0c472b3
fix: image transform order
89d4af3
fix: more imports + loading files
d4f762a
fix: model import
145deb3
fix: masked fill and imports
999d005
change colors
929c70d
adjust README
08233cf
add requirements + gradio demo
e042f04
git attributes
51c3022
Merge pull request #12 from amaiasalvador/master
5eaea16
unverified
Adriana Romero
commited on
Update README.md with correct pytorch version
343ddb1
unverified
Amaia Salvador
commited on
Update requirements.txt with torch and torchvision versions
bf19a93
unverified
Amaia Salvador
commited on
Merge pull request #7 from amaiasalvador/master
da5dc80
unverified
Adriana Romero
commited on
ingredient encoder parameters removed when args.ingrs_only = True
9286344
amaiasalvador
commited on
transformer no longer returns unnecessary attention weights. fix: allow backward when training ingredient decoder
3ab629c
amaiasalvador
commited on
fix: removed perminv_ingrs deprecated argument. Improved instructions.
9f2a462
amaiasalvador
commited on
Initial commit
bd07fad
amaiasalvador
commited on