lmzjms commited on
Commit
b271de2
·
1 Parent(s): 5443283

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -156,11 +156,11 @@ class ConversationBot:
156
  self.llm = OpenAI(temperature=0, openai_api_key=openai_api_key)
157
  # self.t2i = T2I(device="cuda:0")
158
  # self.i2t = ImageCaptioning(device="cuda:0")
159
- self.t2a = T2A(device="cuda:0")
160
- self.tts = TTS(device="cuda:0")
161
  # self.t2s = T2S(device="cuda:0")
162
  # self.i2a = I2A(device="cuda:0")
163
- self.a2t = A2T(device="cuda:0")
164
  # self.asr = ASR(device="cuda:0")
165
  # self.inpaint = Inpaint(device="cuda:0")
166
  #self.tts_ood = TTS_OOD(device="cuda:0")
 
156
  self.llm = OpenAI(temperature=0, openai_api_key=openai_api_key)
157
  # self.t2i = T2I(device="cuda:0")
158
  # self.i2t = ImageCaptioning(device="cuda:0")
159
+ self.t2a = T2A(device="cpu")
160
+ self.tts = TTS(device="cpu")
161
  # self.t2s = T2S(device="cuda:0")
162
  # self.i2a = I2A(device="cuda:0")
163
+ self.a2t = A2T(device="cpu")
164
  # self.asr = ASR(device="cuda:0")
165
  # self.inpaint = Inpaint(device="cuda:0")
166
  #self.tts_ood = TTS_OOD(device="cuda:0")