mirror of
https://github.com/ostris/ai-toolkit.git
synced 2026-04-30 19:21:39 +00:00
Switch to unified text encoder for wan models. Pred for 2.2 14b
This commit is contained in:
15
toolkit/models/loaders/comfy.py
Normal file
15
toolkit/models/loaders/comfy.py
Normal file
@@ -0,0 +1,15 @@
|
||||
import os
|
||||
from typing import List
|
||||
from toolkit.paths import COMFY_MODELS_PATH
|
||||
|
||||
|
||||
def get_comfy_path(comfy_files: List[str]) -> str:
|
||||
"""
|
||||
Get the path to the first existing file in the COMFY_MODELS_PATH.
|
||||
"""
|
||||
if COMFY_MODELS_PATH is not None and comfy_files is not None and len(comfy_files) > 0:
|
||||
for file in comfy_files:
|
||||
file_path = os.path.join(COMFY_MODELS_PATH, file)
|
||||
if os.path.exists(file_path):
|
||||
return file_path
|
||||
return None
|
||||
Reference in New Issue
Block a user