28 lines
663 B
Python
28 lines
663 B
Python
import os
|
|
import hashlib
|
|
|
|
import folder_paths
|
|
|
|
|
|
folder_paths.folder_names_and_paths["AnimateDiff"] = (
|
|
[
|
|
os.path.join(folder_paths.models_dir, "AnimateDiff"),
|
|
os.path.join(os.path.dirname(os.path.abspath(__file__)), "..", "models"),
|
|
],
|
|
folder_paths.supported_pt_extensions,
|
|
)
|
|
|
|
|
|
def get_available_models():
|
|
return folder_paths.get_filename_list("AnimateDiff")
|
|
|
|
|
|
def get_model_path(model_name):
|
|
return folder_paths.get_full_path("AnimateDiff", model_name)
|
|
|
|
|
|
def get_model_hash(file_path):
|
|
with open(file_path, "rb") as f:
|
|
bytes = f.read() # read entire file as bytes
|
|
return hashlib.sha256(bytes).hexdigest()
|