AssertionError when i try to move model to direct ml

[limestackscode]

I cant move model to direct ml but it works with cpu
since this works fine with cpu i assume this is a direct ml issue but i could be wrong
maybe im just doing something wrong here and if thats the case any help would be appreciated

im using torch-directml 0.2.4.dev240815

code that doesnt work :(

import torch
import torch_directml
dml = torch_directml.device()


device = dml



model = torch.hub.load(repo_or_dir="miccunifi/ARNIQA", source="github", model="ARNIQA",
                       regressor_dataset="kadid10k")    # You can choose any of the available datasets
model.eval().to(device)

outputs:
File "C:\Users\AppData\Local\Programs\Python\Python311\Lib\site-packages\torch\nn\modules\module.py", line 830, in _apply
assert isinstance(param, Parameter)
AssertionError

[Lookforworld]

I'm having the same issue, is there any new progress?