TypeError: __init__() got an unexpected keyword argument 'norm_first'

I get this error when trying the new htdemucs with demucs 4.0.0

(demucs) C:\Users\Admin>demucs -n htdemucs D:\input.wav
Traceback (most recent call last):
  File "d:\programas\anaconda\envs\demucs\lib\runpy.py", line 194, in _run_module_as_main
    return _run_code(code, main_globals, None,
  File "d:\programas\anaconda\envs\demucs\lib\runpy.py", line 87, in _run_code
    exec(code, run_globals)
  File "D:\Programas\Anaconda\envs\demucs\Scripts\demucs.exe\__main__.py", line 7, in <module>
  File "d:\programas\anaconda\envs\demucs\lib\site-packages\demucs\separate.py", line 121, in main
    model = get_model_from_args(args)
  File "d:\programas\anaconda\envs\demucs\lib\site-packages\demucs\pretrained.py", line 89, in get_model_from_args
    return get_model(name=args.name, repo=args.repo)
  File "d:\programas\anaconda\envs\demucs\lib\site-packages\demucs\pretrained.py", line 74, in get_model
    model = any_repo.get_model(name)
  File "d:\programas\anaconda\envs\demucs\lib\site-packages\demucs\repo.py", line 148, in get_model
    return self.bag_repo.get_model(name_or_sig)
  File "d:\programas\anaconda\envs\demucs\lib\site-packages\demucs\repo.py", line 130, in get_model
    models = [self.model_repo.get_model(sig) for sig in signatures]
  File "d:\programas\anaconda\envs\demucs\lib\site-packages\demucs\repo.py", line 130, in <listcomp>
    models = [self.model_repo.get_model(sig) for sig in signatures]
  File "d:\programas\anaconda\envs\demucs\lib\site-packages\demucs\repo.py", line 67, in get_model
    return load_model(pkg)
  File "d:\programas\anaconda\envs\demucs\lib\site-packages\demucs\states.py", line 62, in load_model
    model = klass(*args, **kwargs)
  File "d:\programas\anaconda\envs\demucs\lib\site-packages\demucs\states.py", line 146, in __init__
    init(self, *args, **kwargs)
  File "d:\programas\anaconda\envs\demucs\lib\site-packages\demucs\htdemucs.py", line 384, in __init__
    self.crosstransformer = CrossTransformerEncoder(
  File "d:\programas\anaconda\envs\demucs\lib\site-packages\demucs\transformer.py", line 636, in __init__
    self.layers.append(MyTransformerEncoderLayer(**kwargs_classic_encoder))
  File "d:\programas\anaconda\envs\demucs\lib\site-packages\demucs\transformer.py", line 297, in __init__
    super().__init__(
TypeError: __init__() got an unexpected keyword argument 'norm_first'

[HENDRIX-ZT2]

Yes, it’s working with torch 1.13.1. So the version in requirements.txt should be increased to when that argument was added to torch.

[HENDRIX-ZT2]

I’ve investigated this a bit. MyTransformerEncoderLayer wants to pass norm_first=False to torch.nn.TransformerEncoderLayer.

requirements.txt says:

torch>=1.8.1 torchaudio>=0.8

I was running torch 1.9.0, which does not have that argument, but later versions do. Currently updating torch to 1.13.1…