HOTFIX v0.2.2

adds matplotlib to requirements checks responses back for None type images
unconst · Oct 6, 2023 · 3e377b5 · 3e377b5
1 parent 2131539
commit 3e377b5
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/VERSION b/VERSION
@@ -1 +1 @@
-0.2.1
+0.2.2
diff --git a/requirements.txt b/requirements.txt
@@ -9,4 +9,5 @@ fabric @ git+https://github.com/sd-fabric/fabric.git
 omegaconf
 image-reward
 clip @ git+https://github.com/openai/CLIP.git
-imagehash
+imagehash
+matplotlib
diff --git a/validator.py b/validator.py
@@ -425,6 +425,10 @@ async def main():
 
     if not config.validator.allow_nsfw:
         for i, response in enumerate(responses):
+            # delete all none images
+            for j, image in enumerate(response.images):
+                if image is None:
+                    del responses[i].images[j]
             if len(response.images) == 0:
                 continue
             clip_input = processor([bt.Tensor.deserialize(image) for image in response.images], return_tensors="pt").to( DEVICE )