Skip to content

Commit

Permalink
fix numpy version bug
Browse files Browse the repository at this point in the history
  • Loading branch information
RetroCirce committed Apr 10, 2023
1 parent ff23807 commit 349944f
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 18 deletions.
3 changes: 2 additions & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ requires = ["setuptools>=61.0"]
build-backend = "setuptools.build_meta"
[project]
name = "laion_clap"
version = "1.1.2"
version = "1.1.3"
authors = [
{ name="Ke Chen", email="knutchen@ucsd.edu" },
{ name="Yusong Wu" },
Expand All @@ -21,6 +21,7 @@ license = {file = "LICENSE"}
readme = "README.md"
requires-python = ">=3.7"
dependencies = [
"numpy==1.23.5",
"soundfile",
"librosa",
"torchlibrosa",
Expand Down
17 changes: 0 additions & 17 deletions src/laion_clap/hook.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,6 @@
Authors (equal contributions): Ke Chen, Yusong Wu, Tianyu Zhang, Yuchen Hui
Support: LAION
"""
import progressbar
import os
import torch
import librosa
Expand All @@ -17,22 +16,6 @@
import wget
from clap_module.factory import load_state_dict

pbar = None


def show_progress(block_num, block_size, total_size):
global pbar
if pbar is None:
pbar = progressbar.ProgressBar(maxval=total_size)
pbar.start()
downloaded = block_num * block_size
print(downloaded, total_size)
if downloaded < total_size:
pbar.update(downloaded)
else:
pbar.finish()
pbar = None


class CLAP_Module(torch.nn.Module):
def __init__(self, enable_fusion=False, device=None, amodel= 'HTSAT-tiny', tmodel='roberta') -> None:
Expand Down

0 comments on commit 349944f

Please sign in to comment.