diff --git a/modelscope/hub/api.py b/modelscope/hub/api.py index dabc97eb..3b65e471 100644 --- a/modelscope/hub/api.py +++ b/modelscope/hub/api.py @@ -394,7 +394,9 @@ class HubApi: 'WeightsName': aigc_model.weight_filename, 'WeightsSha256': aigc_model.weight_sha256, 'WeightsSize': aigc_model.weight_size, - 'TriggerWords': aigc_model.trigger_words + 'TriggerWords': aigc_model.trigger_words, + 'AigcType': aigc_model.aigc_type, + 'VisionFoundation': aigc_model.base_model_type } else: