Skip to content

Commit

Permalink
removed typos
Browse files Browse the repository at this point in the history
  • Loading branch information
grazder committed Mar 14, 2024
1 parent 04d035c commit 6106c5a
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 5 deletions.
2 changes: 1 addition & 1 deletion torchDF/pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -60,7 +60,7 @@ poethepoet = "^0.21"

[tool.poe.tasks]
install-torch-cuda11 = "python -m pip install torch torchaudio --extra-index-url https://download.pytorch.org/whl/cu118/"
install-torch-cpu = "python -m pip install torch==1.13.1 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cpu/"
install-torch-cpu = "python -m pip install torch==2.1.0 torchaudio==2.1.0 --extra-index-url https://download.pytorch.org/whl/cpu/"
install-eval-utils = "python -m pip install -r requirements_eval.txt"
install-dnsmos-utils = "python -m pip install -r requirements_dnsmos.txt"

Expand Down
5 changes: 1 addition & 4 deletions torchDF/torch_df_streaming_minimal.py
Original file line number Diff line number Diff line change
Expand Up @@ -1094,10 +1094,6 @@ def __init__(
)
self.torch_streaming_model = self.torch_streaming_model.to(device)

erb_norm_state = torch.zeros(self.torch_streaming_model.erb_norm_state_shape)
band_unit_norm_state = torch.zeros(
self.torch_streaming_model.band_unit_norm_state_shape
)
analysis_mem = torch.zeros(self.torch_streaming_model.analysis_mem_shape)
synthesis_mem = torch.zeros(self.torch_streaming_model.synthesis_mem_shape)
rolling_erb_buf = torch.zeros(self.torch_streaming_model.rolling_erb_buf_shape)
Expand All @@ -1124,6 +1120,7 @@ def __init__(
.view(self.torch_streaming_model.erb_norm_state_shape)
.to(torch.float32)
) # float() to fix export issue

band_unit_norm_state = (
torch.linspace(
self.torch_streaming_model.linspace_df[0],
Expand Down

0 comments on commit 6106c5a

Please sign in to comment.