PrunaAI

/

segmind-SSD-1B-turbo-tiny-green-smashed

like 5

Post-training Optimization Tool is deprecated and will be removed in the future. Please use Neural Network Compression Framework instead: https://github.com/openvinotoolkit/nncf
Traceback (most recent call last):
File "/home/ubuntu/pruna/ssd1b.py", line 6, in
smashed_model = PrunaModel.load_model(model_path) # Step (3): load the model.
File "<pruna_engine-0.3.4>/pruna_engine/PrunaModel.py", line 285, in load_model
File "<pruna_engine-0.3.4>/pruna_engine/load_model.py", line 104, in load_model
File "<pruna_engine-0.3.4>/pruna_engine/load_model.py", line 92, in
File "<pruna_engine-0.3.4>/pruna_engine/load_model.py", line 394, in load_optimized_diffusers
File "<pruna_engine-0.3.4>/pruna_engine/load_model.py", line 351, in get_pipeline_type_from_weight_name
NameError: name 'PIPELINE_TYPE' is not defined