summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJen Herting <jen@herting.cc>2024-10-18 12:58:39 -0400
committerGitHub <noreply@github.com>2024-10-18 18:58:39 +0200
commitcfae194fbdb74cd2f98bc6cdfd1acb07cb92cf4c (patch)
treef39f8241cfa438edaa9f47dc3ce071ef9a813b74
parent88c193b83a6e09dd00163f39e4903d626606de33 (diff)
downloadspack-cfae194fbdb74cd2f98bc6cdfd1acb07cb92cf4c.tar.gz
spack-cfae194fbdb74cd2f98bc6cdfd1acb07cb92cf4c.tar.bz2
spack-cfae194fbdb74cd2f98bc6cdfd1acb07cb92cf4c.tar.xz
spack-cfae194fbdb74cd2f98bc6cdfd1acb07cb92cf4c.zip
py-coca-pytorch: new package (#47051)
* py-coca-pytorch: new package * [py-coca-pytorch] black --------- Co-authored-by: Alex C Leute <acl2809@rit.edu>
-rw-r--r--var/spack/repos/builtin/packages/py-coca-pytorch/package.py23
1 files changed, 23 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/py-coca-pytorch/package.py b/var/spack/repos/builtin/packages/py-coca-pytorch/package.py
new file mode 100644
index 0000000000..e759919fd6
--- /dev/null
+++ b/var/spack/repos/builtin/packages/py-coca-pytorch/package.py
@@ -0,0 +1,23 @@
+# Copyright 2013-2024 Lawrence Livermore National Security, LLC and other
+# Spack Project Developers. See the top-level COPYRIGHT file for details.
+#
+# SPDX-License-Identifier: (Apache-2.0 OR MIT)
+
+
+from spack.package import *
+
+
+class PyCocaPytorch(PythonPackage):
+ """CoCa, Contrastive Captioners are Image-Text Foundation Models -
+ Pytorch"""
+
+ homepage = "https://github.com/lucidrains/CoCa-pytorch"
+ pypi = "CoCa-pytorch/CoCa-pytorch-0.1.0.tar.gz"
+
+ license("MIT", checked_by="alex391")
+
+ version("0.1.0", sha256="119c83812d140ad197cf4e992db8c373d908af0bffd0a87015546b6a1cf0a316")
+
+ depends_on("py-setuptools", type="build")
+ depends_on("py-einops@0.4:", type=("build", "run"))
+ depends_on("py-torch@1.6:", type=("build", "run"))