Update app.py

app.py

@@ -55,18 +55,20 @@ def add_extra_model_paths() -> None:
 def import_custom_nodes() -> None:
     import asyncio
     import execution
-    from nodes import init_custom_nodes
+    from nodes import init_extra_nodes
     import server
     loop = asyncio.new_event_loop()
     asyncio.set_event_loop(loop)
     server_instance = server.PromptServer(loop)
     execution.PromptQueue(server_instance)
-    init_custom_nodes()
+    init_extra_nodes()
 
 # Setup ComfyUI and custom nodes
 if not os.path.exists("ComfyUI"):
     subprocess.run(["git", "clone", "https://github.com/comfyanonymous/ComfyUI.git"])
 
+subprocess.run(["sh", "-c", "cd ComfyUI && git checkout `git rev-list -1 --before='2025-03-01' master`"])
+
 custom_node_path = "ComfyUI/custom_nodes/ComfyUI_UltimateSDUpscale"
 if not os.path.exists(custom_node_path):
     subprocess.run(["git", "clone", "https://github.com/ssitu/ComfyUI_UltimateSDUpscale.git", custom_node_path])
@@ -106,7 +108,7 @@ if not os.path.exists(esrgan_x4_path):
 add_comfyui_directory_to_sys_path()
 add_extra_model_paths()
 from folder_paths import add_model_folder_path
-add_model_folder_path("unet", find_path("ComfyUI/models/diffusion_models"))
+add_model_folder_path("unet", find_path("diffusion_models"))
 import_custom_nodes()
 
 from nodes import NODE_CLASS_MAPPINGS