add pix2pix

This commit is contained in:
Qing
2023-01-28 21:13:21 +08:00
parent 205170e1e5
commit 05e82598de
14 changed files with 389 additions and 28 deletions

View File

@@ -11,7 +11,8 @@ AVAILABLE_MODELS = [
"cv2",
"manga",
"sd2",
"paint_by_example"
"paint_by_example",
"pix2pix",
]
AVAILABLE_DEVICES = ["cuda", "cpu", "mps"]