From cc430360c168b6a4a70dc52c8bae18aa0411a44f Mon Sep 17 00:00:00 2001
From: Konstantine Tsafatinos <kosta.tsaf@gmail.com>
Date: Thu, 9 Jan 2025 21:32:02 -0500
Subject: [PATCH] add test txt2txt model to constants

---
 skynet/constants.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/skynet/constants.py b/skynet/constants.py
index a4a7bde..7f325ae 100755
--- a/skynet/constants.py
+++ b/skynet/constants.py
@@ -14,7 +14,7 @@ class Size(msgspec.Struct):
 class ModelDesc(msgspec.Struct):
     short: str
     mem: float
-    size: Size
+    size: Size | None
     tags: list[Literal['txt2img', 'img2img', 'inpaint']]
 
 MODELS: dict[str, ModelDesc] = {
@@ -78,6 +78,12 @@ MODELS: dict[str, ModelDesc] = {
         size=Size(w=512, h=512),
         tags=['txt2img']
     ),
+    'microsoft/DialoGPT-small': ModelDesc(
+        short='dgpt-sm',
+        mem=6,
+        size=None,
+        tags=['txt2txt']
+    ),
     'diffusers/stable-diffusion-xl-1.0-inpainting-0.1': ModelDesc(
         short='stablexl-inpainting',
         mem=8.3,