Update mmaudio/ext/bigvgan_v2/bigvgan.py
Browse files
mmaudio/ext/bigvgan_v2/bigvgan.py
CHANGED
|
@@ -362,13 +362,13 @@ class BigVGAN(
|
|
| 362 |
cls,
|
| 363 |
*,
|
| 364 |
model_id: str,
|
| 365 |
-
revision: str,
|
| 366 |
-
cache_dir: str,
|
| 367 |
-
force_download: bool,
|
| 368 |
-
proxies: Optional[Dict],
|
| 369 |
-
resume_download: bool,
|
| 370 |
-
local_files_only: bool,
|
| 371 |
-
token: Union[str, bool, None],
|
| 372 |
map_location: str = "cpu", # Additional argument
|
| 373 |
strict: bool = False, # Additional argument
|
| 374 |
use_cuda_kernel: bool = False,
|
|
@@ -436,4 +436,4 @@ class BigVGAN(
|
|
| 436 |
model.remove_weight_norm()
|
| 437 |
model.load_state_dict(checkpoint_dict["generator"])
|
| 438 |
|
| 439 |
-
return model
|
|
|
|
| 362 |
cls,
|
| 363 |
*,
|
| 364 |
model_id: str,
|
| 365 |
+
revision: Optional[str] = None,
|
| 366 |
+
cache_dir: Optional[str] = None,
|
| 367 |
+
force_download: bool = False,
|
| 368 |
+
proxies: Optional[Dict] = None,
|
| 369 |
+
resume_download: bool = True,
|
| 370 |
+
local_files_only: bool = False,
|
| 371 |
+
token: Union[str, bool, None] = None,
|
| 372 |
map_location: str = "cpu", # Additional argument
|
| 373 |
strict: bool = False, # Additional argument
|
| 374 |
use_cuda_kernel: bool = False,
|
|
|
|
| 436 |
model.remove_weight_norm()
|
| 437 |
model.load_state_dict(checkpoint_dict["generator"])
|
| 438 |
|
| 439 |
+
return model
|