Skip to content

Commit

Permalink
delete annotation
Browse files Browse the repository at this point in the history
  • Loading branch information
ZaymeShaw committed Jul 27, 2024
1 parent 9e96be7 commit 0820f3c
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion ChatTTS/model/tokenizer.py
Original file line number Diff line number Diff line change
Expand Up @@ -185,7 +185,6 @@ def _decode_prompt(prompt: str) -> torch.Tensor:
@staticmethod
@torch.no_grad()
def _encode_prompt(prompt: torch.Tensor) -> str:
# arr: np.ndarray = prompt.to(dtype=torch.uint16, device="cpu").numpy()
arr: np.ndarray = prompt.to(device="cpu").numpy().astype(np.uint16)
shp = arr.shape
assert len(shp) == 2, "prompt must be a 2D tensor"
Expand Down

0 comments on commit 0820f3c

Please sign in to comment.