cowtopia

Paused

amarax commited on Apr 5, 2023

Commit

23a4337

1 Parent(s): 96cad2d

Removed Image Editing for prompt stability

Files changed (1) hide show

app.py CHANGED Viewed

@@ -146,7 +146,7 @@ class ConversationBot:
 bot = ConversationBot({'Text2Image': 'cuda:0',
                        'ImageCaptioning': 'cuda:0',
-                       'ImageEditing': 'cuda:0',
                        'VisualQuestionAnswering': 'cuda:0',
                     #    'Image2Canny': 'cpu',
                     #    'CannyText2Image': 'cuda:0',

 bot = ConversationBot({'Text2Image': 'cuda:0',
                        'ImageCaptioning': 'cuda:0',
+                    #    'ImageEditing': 'cuda:0',
                        'VisualQuestionAnswering': 'cuda:0',
                     #    'Image2Canny': 'cpu',
                     #    'CannyText2Image': 'cuda:0',