KeyError: 'model_file' using safety.sensitive model

[EternalDusk]

(Using Ubuntu on WSL, other models work so I don't believe WSL is the issue. OpenChat will not load on windows.)

from openchat import OpenChat OpenChat(model="safety.sensitive", device="cpu")

                              ____   ____   ______ _   __   ______ __  __ ___   ______
                             / __ \ / __ \ / ____// | / /  / ____// / / //   | /_  __/
                            / / / // /_/ // __/  /  |/ /  / /    / /_/ // /| |  / /
                           / /_/ // ____// /___ / /|  /  / /___ / __  // ___ | / /
                           \____//_/    /_____//_/ |_/   \____//_/ /_//_/  |_|/_/

                                               ... LOADING ...

Traceback (most recent call last): File "", line 1, in File "/home/dusk/.local/lib/python3.8/site-packages/openchat/openchat.py", line 24, in init self.agent = self.create_agent_by_name( File "/home/dusk/.local/lib/python3.8/site-packages/openchat/openchat.py", line 87, in create_agent_by_name return SensitiveAgent(name, device, maxlen) File "/home/dusk/.local/lib/python3.8/site-packages/openchat/agents/safety.py", line 111, in init model=create_agent_from_opt_file(option), File "/home/dusk/.local/lib/python3.8/site-packages/parlai/core/agents.py", line 361, in create_agent_from_opt_file model_file = opt['model_file'] KeyError: 'model_file'

I don't know quite what is causing this, any help is appreciated!

[EternalDusk]

Tested this again on a clean Ubuntu installation, safety models are not working. Same error as before.