Hugging Face's logo

Spaces:
wffcyrus
/
ChatTTS-Story-Telling
Running

App Files Community
fcyai commited on
Commit
bb02bb0
•
1 Parent(s): 53f5d2c

device

Browse files
Files changed (1) hide show
  1. ChatTTS/model/gpt.py +1 -1
ChatTTS/model/gpt.py CHANGED
@@ -83,7 +83,7 @@ class GPT_warpper(nn.Module):
83
  return model.to(device)
84
 
85
  def get_emb(self, input_ids, text_mask):
86
-
87
  emb_text = self.emb_text(input_ids[text_mask][:, 0].to(self.device_gpt))
88
 
89
  text_mask_inv = ~text_mask
 
83
  return model.to(device)
84
 
85
  def get_emb(self, input_ids, text_mask):
86
+ print(f'get_emb gpt_device:{self.device_gpt}, device:{self.device},')
87
  emb_text = self.emb_text(input_ids[text_mask][:, 0].to(self.device_gpt))
88
 
89
  text_mask_inv = ~text_mask