Skip to content

Commit

Permalink
Update pig.py
Browse files Browse the repository at this point in the history
  • Loading branch information
calcuis authored Jan 20, 2025
1 parent 6cd2efd commit 37a46ac
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions pig.py
Original file line number Diff line number Diff line change
Expand Up @@ -285,7 +285,7 @@ def get_folder_names_and_paths(key, targets=[]):
f'Unknown file list already present on key {key}: {base}')
get_folder_names_and_paths('model_gguf', ['diffusion_models', 'unet'])
get_folder_names_and_paths('clip_gguf', ['text_encoders', 'clip'])
CUS_ARCH_LIST = {'flux', 'sd1', 'sdxl', 'sd3', 'aura', 'mochi', 'ltxv', 'hyvid', 'cosmos'}
PIG_ARCH_LIST = {'flux', 'sd1', 'sdxl', 'sd3', 'aura', 'mochi', 'ltxv', 'hyvid', 'cosmos'}
TXT_ARCH_LIST = {'t5', 't5encoder', 'llama'}
def get_orig_shape(reader, tensor_name):
field_key = f'comfy.gguf.orig_shape.{tensor_name}'
Expand Down Expand Up @@ -326,7 +326,7 @@ def load_gguf_sd(path, handle_prefix='model.diffusion_model.',
f'Bad type for GGUF general.architecture key: expected string, got {arch_field.types!r}'
)
arch_str = str(arch_field.parts[arch_field.data[-1]], encoding='utf-8')
if arch_str not in CUS_ARCH_LIST and arch_str not in TXT_ARCH_LIST:
if arch_str not in PIG_ARCH_LIST and arch_str not in TXT_ARCH_LIST:
raise ValueError(
f'Unexpected architecture type in GGUF file, expected one of flux, sd1-3/sdxl, ltxv, hyvid, t5encoder, etc. but got {arch_str!r}'
)
Expand Down Expand Up @@ -511,7 +511,7 @@ class DualClipLoaderGGUF(ClipLoaderGGUF):
def INPUT_TYPES(s):
file_options = s.get_filename_list(),
return {'required': {'clip_name1': file_options, 'clip_name2':
file_options, 'type': (('sdxl', 'sd3', 'flux'),)}}
file_options, 'type': (('sdxl', 'sd3', 'flux', 'hunyuan_video'),)}}
TITLE = 'GGUF DualCLIPLoader'
def load_clip(self, clip_name1, clip_name2, type):
clip_path1 = folder_paths.get_full_path('clip', clip_name1)
Expand Down

0 comments on commit 37a46ac

Please sign in to comment.