han-xudong commited on
Commit ·
e7ce8c7
1
Parent(s): 930efdf
modified: modeling.py
Browse files- modeling.py +1 -0
modeling.py
CHANGED
|
@@ -104,6 +104,7 @@ class MVAE(PreTrainedModel):
|
|
| 104 |
|
| 105 |
@classmethod
|
| 106 |
def from_pretrained(cls, pretrained_model_name_or_path, *args, **kwargs):
|
|
|
|
| 107 |
prosoro_type = kwargs.pop("prosoro_type", None)
|
| 108 |
|
| 109 |
print(f"Loading model from {pretrained_model_name_or_path} with prosoro_type={prosoro_type}")
|
|
|
|
| 104 |
|
| 105 |
@classmethod
|
| 106 |
def from_pretrained(cls, pretrained_model_name_or_path, *args, **kwargs):
|
| 107 |
+
print(kwargs)
|
| 108 |
prosoro_type = kwargs.pop("prosoro_type", None)
|
| 109 |
|
| 110 |
print(f"Loading model from {pretrained_model_name_or_path} with prosoro_type={prosoro_type}")
|