Commit Graph

355 Commits

Author SHA1 Message Date
kuprel
d4af693133 updated colab to load model once and generate multiple times 2022-06-29 09:43:42 -04:00
Brett Kuprel
ed91ab4a30 refactored to load models once and run multiple times 2022-06-29 09:42:12 -04:00
kuprel
1ef9b0b929 added mega to colab 2022-06-29 06:44:07 -04:00
kuprel
ae0411dbcc Cleanup 2022-06-29 06:25:52 -04:00
kuprel
0fe3e2d2b9
Merge pull request #28 from interfect/torch-logical-cores
Use all logical cores in Torch mode
2022-06-28 22:57:02 -04:00
Adam Novak
28c812c832 Use all logical cores in Torch mode 2022-06-28 22:26:51 -04:00
kuprel
97a55f169c Updated colab to use torch+cuda 2022-06-28 22:12:36 -04:00
Brett Kuprel
1fbb209623 fixed bug with cuda in detokenizer 2022-06-28 22:02:35 -04:00
Brett Kuprel
764b0bc685 cuda in detokenizer from previous commit broke colab flax model, fixed 2022-06-28 21:36:48 -04:00
Brett Kuprel
d846cab1b6 Merge branch 'main' of https://github.com/kuprel/min-dalle 2022-06-28 21:28:50 -04:00
Brett Kuprel
17c96fe110 works with cuda 2022-06-28 21:28:36 -04:00
kuprel
0af6b1731f
Update setup.sh
removed `git lfs` dependency
2022-06-28 21:18:24 -04:00
kuprel
6f0be73579
Merge pull request #23 from andrewginns/main
Simplified requirements
2022-06-28 21:16:00 -04:00
kuprel
b8c4173181
Merge pull request #24 from TheFutureGadgetsLab/main
Fixed disabling of gradients in the torch code
2022-06-28 21:10:35 -04:00
Haydn Jones
a3a247e6ec Fixed disabling of gradients in the torch code 2022-06-28 17:57:44 -06:00
Andrew Ginns
3c0c897977 Add flax model curl 2022-06-28 22:01:26 +01:00
Andrew Ginns
3af89066e8 Simplified requirements:
* No wandb login
* wandb install as part of requirements.txt
2022-06-28 21:34:57 +01:00
Brett Kuprel
9d6b6dcc92 previous commit broke flax model, fixed now 2022-06-28 12:54:58 -04:00
Evan Pratten
2bb4759f61
Explicitly call wandb as a python module.
This change allows wandb to be called on systems that do not automatically add python modules to $PATH
2022-06-28 12:51:24 -04:00
Brett Kuprel
5aa6fe49bf use cuda if available 2022-06-28 12:47:11 -04:00
Brett Kuprel
8544f59576 use cuda if available 2022-06-28 12:38:31 -04:00
Brett Kuprel
aef24ea157 torch.no_grad(), cleanup 2022-06-28 12:16:44 -04:00
kuprel
1e2649148b
Merge pull request #19 from jedahan/patch-2
Support pip/pip3 binary names for setup
2022-06-28 11:35:20 -04:00
kuprel
57127ac616
Merge branch 'main' into patch-2 2022-06-28 11:34:46 -04:00
kuprel
6baeeec7f1
Merge pull request #17 from jedahan/patch-1
Exit script immediately if any command fails
2022-06-28 11:31:42 -04:00
Jonathan Dahan
fd0e808454
Support pip/pip3 binary names for setup 2022-06-28 11:20:23 -04:00
Jonathan Dahan
63a65372b0
Exit script immediately if any command fails 2022-06-28 11:13:10 -04:00
kuprel
7478c5ffa3 Simplified colab 2022-06-28 11:05:59 -04:00
kuprel
cbb8d4156f Simplified colab 2022-06-28 11:01:31 -04:00
kuprel
407ea96865
Merge pull request #14 from rupa/main
fix requirements.txt
2022-06-28 09:25:58 -04:00
kuprel
6dbb5d7a7b Login to weights and biases anonymously in colab 2022-06-28 09:19:52 -04:00
rupa
647eb950c2 fix requirements.txt 2022-06-28 09:18:23 -04:00
Brett Kuprel
34df2b97df previous commit broke colab example, so adjusting flax requirement to 0.4.2 for now 2022-06-28 08:04:08 -04:00
Brett Kuprel
38ebe54a38 works with latest flax version 0.5.2 now 2022-06-28 07:12:29 -04:00
kuprel
50e1e74c4a
Merge pull request #6 from osanseviero/patch-1
Update README.md
2022-06-28 06:28:21 -04:00
kuprel
332677b032 Fixed bug in colab notebook setup process 2022-06-28 06:09:52 -04:00
Omar Sanseviero
efa40ab321
Update README.md 2022-06-28 09:25:56 +02:00
Brett Kuprel
6260252348 update readme 2022-06-27 22:08:29 -04:00
kuprel
2ad7009a16
Update README.md 2022-06-27 21:11:27 -04:00
Brett Kuprel
6a068651e5 updated colab 2022-06-27 20:59:39 -04:00
kuprel
8ea0482d03 Cleaned up notebook 2022-06-27 20:58:17 -04:00
Brett Kuprel
24d8e29ef2 readme formatting 2022-06-27 17:28:40 -04:00
Brett Kuprel
5444c73e3f Merge branch 'main' of https://github.com/kuprel/min-dalle
commited from colab separately
2022-06-27 17:26:53 -04:00
Brett Kuprel
8363495f0a updated readme 2022-06-27 17:26:05 -04:00
kuprel
9d6bacac14 Created using Colaboratory 2022-06-27 17:23:30 -04:00
Brett Kuprel
a014dccc05 fixed an issue with argument parser 2022-06-27 16:49:42 -04:00
Brett Kuprel
e7001f063c simplified 2022-06-27 15:46:04 -04:00
Brett Kuprel
18e6a9852f license and cleanup 2022-06-27 14:34:10 -04:00
Brett Kuprel
32b7aa196b readme 2022-06-27 13:51:48 -04:00
Brett Kuprel
56b24494ae examples 2022-06-27 13:39:00 -04:00