Spaces:

AIGC-Audio
/

Make_An_Audio

Runtime error

lmzjms commited on May 22, 2023

Commit

5316053

1 Parent(s): 85c78a5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -289,13 +289,13 @@ class ConversationBot:
 if __name__ == '__main__':
     bot = ConversationBot({'ImageCaptioning': 'cuda:0',
-                           'T2A': 'cuda:0',
-                           'I2A': 'cuda:0',
                            'TTS': 'cpu',
                            'T2S': 'cpu',
                            'ASR': 'cuda:0',
                            'A2T': 'cpu',
-                           'Inpaint': 'cuda:0',
                            'SoundDetection': 'cpu',
                            'Binaural': 'cuda:0',
                            'SoundExtraction': 'cuda:0',

 if __name__ == '__main__':
     bot = ConversationBot({'ImageCaptioning': 'cuda:0',
+                           #'T2A': 'cuda:0',
+                           #'I2A': 'cuda:0',
                            'TTS': 'cpu',
                            'T2S': 'cpu',
                            'ASR': 'cuda:0',
                            'A2T': 'cpu',
+                           #'Inpaint': 'cuda:0',
                            'SoundDetection': 'cpu',
                            'Binaural': 'cuda:0',
                            'SoundExtraction': 'cuda:0',