seawolf2357 commited on
Commit
1c62477
·
verified ·
1 Parent(s): 4cca498

Update mmaudio/ext/bigvgan_v2/bigvgan.py

Browse files
Files changed (1) hide show
  1. mmaudio/ext/bigvgan_v2/bigvgan.py +8 -8
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