21 lines
705 B
Python
21 lines
705 B
Python
import os
|
|
|
|
def get_comfyui_base_path():
|
|
# Assuming the script is in the standard directory structure
|
|
return os.path.abspath(os.path.join(os.path.dirname(__file__), "..", ".."))
|
|
|
|
def get_folder_paths(folder_type):
|
|
base_path = get_comfyui_base_path()
|
|
|
|
if folder_type == "diffusers":
|
|
return [os.path.join(base_path, "models", "diffusers")]
|
|
if folder_type == "embeddings":
|
|
return [os.path.join(base_path, "models", "embeddings")]
|
|
|
|
return []
|
|
|
|
# Example usage:
|
|
if __name__ == "__main__":
|
|
print(get_folder_paths("diffusers")) # Should print the path to the diffusers models
|
|
print(get_folder_paths("embeddings")) # Should print the path to the embeddings
|