Commit Graph

207 Commits

Author SHA1 Message Date
Brett Kuprel 256fd1aab0 empty cache if cuda available 2022-07-04 07:21:54 -04:00
Brett Kuprel 9602055eaa Merge pull request #57 from w4ffl35/bugfix-for-mega-model
Fixes #16 - mega model running out of memory
2022-07-04 07:16:39 -04:00
w4ffl35 7d549505da Fixes #16 - mega model running out of memory
Prior to this fix the mega model will (more often than not) fail when running in succession.

Clearing the cache seems to fix the issue.
2022-07-04 00:26:17 -06:00
Brett Kuprel 6b433d5ba6 update colab 2022-07-03 18:40:27 -04:00
Brett Kuprel 53741154c7 update readme 2022-07-03 17:29:53 -04:00
Brett Kuprel aa08f7f93c update readme 2022-07-03 16:49:38 -04:00
Brett Kuprel ec14eeba62 convert pngs to jpgs 2022-07-03 16:30:36 -04:00
Brett Kuprel 2661074589 update readme 2022-07-03 16:26:49 -04:00
Brett Kuprel 6ce60179ff update cog.yaml 2022-07-03 16:16:23 -04:00
Brett Kuprel 948e05c0dd update readme 2022-07-03 15:40:58 -04:00
Brett Kuprel e20c4a8c9f updated colab 2022-07-03 15:35:33 -04:00
Brett Kuprel a5004efad0 update readme 2022-07-03 13:35:36 -04:00
Brett Kuprel b939a1e5ac update cog.yaml 2022-07-03 12:20:03 -04:00
Brett Kuprel f0f9c5e83e update replicate predict.py 2022-07-03 12:14:24 -04:00
Brett Kuprel 8d808599d3 h100 stat tbd 2022-07-03 11:53:45 -04:00
Brett Kuprel 229af7233d h100 stat tbd 2022-07-03 11:34:58 -04:00
Brett Kuprel 107213b28b 3x3 only works on colab if P100 is allocated 2022-07-03 11:28:28 -04:00
Brett Kuprel f1bb562ac1 update readme 2022-07-03 11:22:52 -04:00
Brett Kuprel 134fdaf76b add nvidia-smi to notebook 2022-07-03 11:17:37 -04:00
Brett Kuprel 9bdba57195 update replicate 2022-07-02 16:36:51 -04:00
Brett Kuprel 9e146e6675 update readme 2022-07-02 14:56:42 -04:00
Brett Kuprel cbee448f11 update notebook 2022-07-02 14:54:18 -04:00
Brett Kuprel a22d510270 update readme 2022-07-02 13:06:48 -04:00
Brett Kuprel d8a0cbcaf5 update readme 2022-07-02 13:06:03 -04:00
Brett Kuprel 1351478ee5 update readme 2022-07-02 12:55:59 -04:00
Brett Kuprel 05d9d5f07f update readme 2022-07-02 12:34:11 -04:00
Brett Kuprel bf1acab78a update readme 2022-07-02 12:32:54 -04:00
Brett Kuprel ec37a2dafe update replicate cog yaml file 2022-07-02 12:02:55 -04:00
Brett Kuprel e8dac15810 update replicate max grid size 2022-07-02 11:59:26 -04:00
Brett Kuprel 31fd7c231d updated replicate cog.yaml python packages 2022-07-02 11:23:32 -04:00
Brett Kuprel 849e720313 add default arguments to replicate 2022-07-02 10:46:56 -04:00
Brett Kuprel 465b2428de update replicate files 2022-07-02 10:35:27 -04:00
Brett Kuprel 9cde5b1a60 update replicate files 2022-07-02 10:32:17 -04:00
Brett Kuprel e6f1c3ef4c update replicate files 2022-07-02 10:05:16 -04:00
Brett Kuprel a91a259293 Merge pull request #49 from raphant/patch-1
Update README.md to include all required third-party dependencies
2022-07-02 09:32:25 -04:00
Brett Kuprel 8b77428102 Merge branch 'main' into patch-1 2022-07-02 09:32:15 -04:00
Brett Kuprel 9e7d6edc30 updated colab example 2022-07-02 09:31:20 -04:00
Brett Kuprel f8cae82b93 update readme 2022-07-02 09:26:30 -04:00
Brett Kuprel 20a91e8714 add grid-size example to readme 2022-07-02 09:23:59 -04:00
Brett Kuprel 72cd1501f2 update replicate file 2022-07-02 09:20:42 -04:00
Brett Kuprel e087c53482 update readme 2022-07-02 09:13:05 -04:00
Brett Kuprel 313635e914 fixed typing error for older python versions 2022-07-02 09:06:22 -04:00
Brett Kuprel 2dadfdfb31 generate 3x3 grid in colab on standard GPU runtime 2022-07-02 09:04:13 -04:00
Raphael 0d2d4a6331 Update requirements.txt
Add required third-party libraries
2022-07-02 08:52:05 -04:00
Brett Kuprel 1eb56737d8 added grid_size parameter to generate a grid of images 2022-07-02 08:45:49 -04:00
Raphael 58a954f777 Update README.md 2022-07-02 08:38:45 -04:00
Brett Kuprel e0386f991c removed typing_extensions requirement 2022-07-02 06:34:47 -04:00
Brett Kuprel 4b632555b5 models_root command line argument 2022-07-01 23:39:48 -04:00
Brett Kuprel 21c633b56e update readme 2022-07-01 20:27:29 -04:00
Brett Kuprel 9cbaa83b05 update readme 2022-07-01 20:25:49 -04:00