Skip to content

Commit 3600033

Browse files
committed
Deleted dead code, removed unneeded dependency, bumped version for release
1 parent ac1076d commit 3600033

File tree

3 files changed

+158
-254
lines changed

3 files changed

+158
-254
lines changed

audio_separator/separator/uvr_lib_v5/demucs/states.py

Lines changed: 0 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,6 @@
1515
from pathlib import Path
1616
import warnings
1717

18-
from omegaconf import OmegaConf
1918
from diffq import DiffQuantizer, UniformQuantizer, restore_quantized_state
2019
import torch
2120

@@ -101,14 +100,6 @@ def save_with_checksum(content, path):
101100
path.write_bytes(buf.getvalue())
102101

103102

104-
def serialize_model(model, training_args, quantizer=None, half=True):
105-
args, kwargs = model._init_args_kwargs
106-
klass = model.__class__
107-
108-
state = get_state(model, quantizer, half)
109-
return {"klass": klass, "args": args, "kwargs": kwargs, "state": state, "training_args": OmegaConf.to_container(training_args, resolve=True)}
110-
111-
112103
def copy_state(state):
113104
return {k: v.cpu().clone() for k, v in state.items()}
114105

0 commit comments

Comments
 (0)