lmzjms commited on
Commit
212bdfd
·
1 Parent(s): ba9b7b5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -160,7 +160,7 @@ class ConversationBot:
160
  self.llm = OpenAI(temperature=0, openai_api_key=openai_api_key)
161
  # self.t2i = T2I(device="cuda:0")
162
  # self.i2t = ImageCaptioning(device="cuda:0")
163
- self.t2a = T2A(device="cpu")
164
  self.tts = TTS(device="cpu")
165
  self.t2s = T2S(device="cpu")
166
  # self.i2a = I2A(device="cpu")
 
160
  self.llm = OpenAI(temperature=0, openai_api_key=openai_api_key)
161
  # self.t2i = T2I(device="cuda:0")
162
  # self.i2t = ImageCaptioning(device="cuda:0")
163
+ self.t2a = T2A(device="cuda:0")
164
  self.tts = TTS(device="cpu")
165
  self.t2s = T2S(device="cpu")
166
  # self.i2a = I2A(device="cpu")