VidMuse_CVPR
Browse files- processor.py +2 -1
processor.py
CHANGED
|
@@ -5,8 +5,9 @@ class VidMuseConfig(AutoConfig):
|
|
| 5 |
model_type = "simple_processor" # 类属性,不是构造函数参数
|
| 6 |
|
| 7 |
def __init__(self, message="Hello from VidMuseConfig!", **kwargs):
|
| 8 |
-
# 从 kwargs 中移除 model_type,避免传递给父类
|
| 9 |
kwargs.pop("model_type", None)
|
|
|
|
| 10 |
super().__init__(**kwargs) # 传递其他参数给父类
|
| 11 |
self.message = message
|
| 12 |
|
|
|
|
| 5 |
model_type = "simple_processor" # 类属性,不是构造函数参数
|
| 6 |
|
| 7 |
def __init__(self, message="Hello from VidMuseConfig!", **kwargs):
|
| 8 |
+
# 从 kwargs 中移除 model_type 和 auto_map,避免传递给父类
|
| 9 |
kwargs.pop("model_type", None)
|
| 10 |
+
kwargs.pop("auto_map", None)
|
| 11 |
super().__init__(**kwargs) # 传递其他参数给父类
|
| 12 |
self.message = message
|
| 13 |
|