fix error

This commit is contained in:
raysers
2024-10-18 02:14:15 -07:00
parent 1f4c56e722
commit fd92dba2ef
+1
View File
@@ -47,6 +47,7 @@ class MfluxModelsLoader:
@classmethod
def get_model_paths(cls):
mflux_dir = os.path.join(models_dir, "Mflux")
Path(mflux_dir).mkdir(parents=True, exist_ok=True)
return [p.name for p in Path(mflux_dir).iterdir() if p.is_dir()]
def load(self, model_name):