WIP create new class to add new models more easily

toolkit/util/get_model.py

@@ -0,0 +1,9 @@
+from toolkit.stable_diffusion_model import StableDiffusion
+from toolkit.config_modules import ModelConfig
+
+def get_model_class(config: ModelConfig):
+    if config.arch == "wan21":
+        from toolkit.models.wan21 import Wan21
+        return Wan21
+    else:
+        return StableDiffusion