Stable-X/yoso-normal-v1-8-1 · Missing fields in model

When running the model with torch.hub.load with torch.hub.load("Stable-X/StableNormal", [...], we endup having this error :
ValueError: Pipeline <class 'stablenormal.pipeline_yoso_normal.YOSONormalsPipeline'> expected {'text_encoder', 'tokenizer', 'safety_checker', 'image_encoder', 'scheduler', 'unet', 'vae', 'feature_extractor', 'controlnet'}, but only {'safety_checker', 'unet', 'controlnet', 'vae'} were passed.
But it works well with the fork hugoycj/StableNormal. Could you consider merge the changes ?

Stable-X
/

yoso-normal-v1-8-1

Missing fields in model_index.json