Update modeling_chatglm.py
Browse files- modeling_chatglm.py +1 -1
modeling_chatglm.py
CHANGED
@@ -731,7 +731,7 @@ class Embedding(torch.nn.Module):
|
|
731 |
|
732 |
|
733 |
class ChatGLMModel(ChatGLMPreTrainedModel):
|
734 |
-
def __init__(self, config: ChatGLMConfig, device=None, empty_init=
|
735 |
super().__init__(config)
|
736 |
if empty_init:
|
737 |
init_method = skip_init
|
|
|
731 |
|
732 |
|
733 |
class ChatGLMModel(ChatGLMPreTrainedModel):
|
734 |
+
def __init__(self, config: ChatGLMConfig, device=None, empty_init=False):
|
735 |
super().__init__(config)
|
736 |
if empty_init:
|
737 |
init_method = skip_init
|