Spaces:

AIGC-Audio
/

Make_An_Audio

Running on A10G

lmzjms commited on May 22, 2023

Commit

c1fded2

1 Parent(s): 85c8fb1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -205,13 +205,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',