This commit is contained in:
enhuiz 2023-01-12 19:55:04 +08:00
parent 74b9dc5068
commit 886c11de46

View File

@ -8,7 +8,7 @@ def get_model(name):
if name.startswith("ar"):
Model = AR
elif str().startswith("nar"):
elif name.startswith("nar"):
Model = NAR
else:
raise ValueError("Model name should start with AR or NAR.")