140 lines
2.0 KiB
Python
140 lines
2.0 KiB
Python
load("@rules_python//python:defs.bzl", "py_library", "py_test")
|
|
|
|
package(default_visibility=["//visibility:public"])
|
|
|
|
py_library(
|
|
name="config",
|
|
srcs=["config.py"],
|
|
deps=[
|
|
":logger",
|
|
":constants",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name="constants",
|
|
srcs=["constants.py"],
|
|
)
|
|
|
|
py_library(
|
|
name="database",
|
|
srcs=["database.py"],
|
|
deps=[
|
|
":logger",
|
|
":times",
|
|
":images",
|
|
":constants",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name="external",
|
|
srcs=["external.py"],
|
|
deps=[
|
|
":logger",
|
|
":config",
|
|
":constants",
|
|
":images",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name="envvar",
|
|
srcs=["envvar.py"],
|
|
)
|
|
|
|
py_test(
|
|
name="envvar_test",
|
|
srcs=["envvar_test.py"],
|
|
deps=[":envvar"],
|
|
)
|
|
|
|
py_library(
|
|
name="images",
|
|
srcs=["images.py"],
|
|
)
|
|
|
|
py_library(
|
|
name="logger",
|
|
srcs=["logger.py"],
|
|
)
|
|
|
|
py_test(
|
|
name="logger_test",
|
|
srcs=["logger_test.py"],
|
|
deps=[":logger"],
|
|
)
|
|
|
|
py_library(
|
|
name="memory",
|
|
srcs=["memory.py"],
|
|
)
|
|
|
|
py_library(
|
|
name="model",
|
|
srcs=["model.py"],
|
|
deps=[
|
|
":constants",
|
|
":memory",
|
|
":logger",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name="text2img",
|
|
srcs=["text2img.py"],
|
|
deps=[
|
|
":constants",
|
|
":config",
|
|
":logger",
|
|
":images",
|
|
":memory",
|
|
":model",
|
|
":times",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name="img2img",
|
|
srcs=["img2img.py"],
|
|
deps=[
|
|
":constants",
|
|
":config",
|
|
":logger",
|
|
":images",
|
|
":memory",
|
|
":model",
|
|
":times",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name="inpainting",
|
|
srcs=["inpainting.py"],
|
|
deps=[
|
|
":constants",
|
|
":config",
|
|
":logger",
|
|
":images",
|
|
":memory",
|
|
":model",
|
|
":times",
|
|
],
|
|
)
|
|
|
|
py_library(
|
|
name="times",
|
|
srcs=["times.py"],
|
|
)
|
|
|
|
py_test(
|
|
name="times_test",
|
|
srcs=["times_test.py"],
|
|
deps=[":times"],
|
|
)
|
|
|
|
py_library(
|
|
name="translator",
|
|
srcs=["translator.py"],
|
|
)
|