Ean Garvey 6cf5564c84 Remove "gpu" device alias and migrate to using "cuda" for NVIDIA GPU. (#325) ...

* Replace instances of "gpu" alias for devices with "cuda"

2022-09-13 01:16:56 -05:00

..

diffusion

Remove "gpu" device alias and migrate to using "cuda" for NVIDIA GPU. (#325)

2022-09-13 01:16:56 -05:00

__init__.py

[WEB] Add v_diffusion model in the shark web (#306)

2022-09-01 06:34:51 -07:00

albert_maskfill.py

[WEB] Cache the compiled module.

2022-09-01 23:08:08 +05:30

resnet50.py

[WEB] Cache the compiled module.

2022-09-01 23:08:08 +05:30