model_name = "pyramid_flux", - error
Olá,
Colocando o model_name = "pyramid_flux" , apresentou o erro abaixo. Saberia me dizer o que é?
'''
Using temporal causal attention
Traceback (most recent call last):
File "D:\GPT\piramid\pyramid.venv\ExecModel.py", line 21, in
model = PyramidDiTForVideoGeneration(
File "D:\GPT/piramid/pyramid/Pyramid-Flow\pyramid_dit\pyramid_dit_for_video_gen_pipeline.py", line 141, in init
self.dit = build_pyramid_dit(
File "D:\GPT/piramid/pyramid/Pyramid-Flow\pyramid_dit\pyramid_dit_for_video_gen_pipeline.py", line 73, in build_pyramid_dit
dit = PyramidFluxTransformer.from_pretrained(
File "D:\GPT\piramid\pyramid.venv\lib\site-packages\huggingface_hub\utils_validators.py", line 114, in _inner_fn
return fn(*args, **kwargs)
File "D:\GPT\piramid\pyramid.venv\lib\site-packages\diffusers\models\modeling_utils.py", line 848, in from_pretrained
raise ValueError(
ValueError: Cannot load <class 'pyramid_dit.flux_modules.modeling_pyramid_flux
'''