Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
  • Loading branch information
AnyaCoder committed Sep 11, 2024
2 parents 173bc8a + 254cbb5 commit a4c8063
Show file tree
Hide file tree
Showing 3 changed files with 1 addition and 3 deletions.
1 change: 0 additions & 1 deletion fish_speech/webui/manage.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,6 @@
import datetime
import html
import json

import platform
import shutil
import signal
Expand Down
2 changes: 1 addition & 1 deletion tools/llama/generate.py
Original file line number Diff line number Diff line change
Expand Up @@ -238,7 +238,7 @@ def generate(
T = prompt.size(1)

device, dtype = prompt.device, prompt.dtype

codebook_dim = 1 + model.config.num_codebooks
# create an empty tensor of the expected final shape and fill in the current tokens
empty = torch.empty((codebook_dim, max_new_tokens), dtype=dtype, device=device)
Expand Down
1 change: 0 additions & 1 deletion tools/webui.py
Original file line number Diff line number Diff line change
Expand Up @@ -383,7 +383,6 @@ def build_app():
fn=normalize_text, inputs=[text, if_refine_text], outputs=[refined_text]
)


# # Submit
generate.click(
inference_wrapper,
Expand Down

0 comments on commit a4c8063

Please sign in to comment.