Update modeling_diva.py
Browse files- modeling_diva.py +5 -1
modeling_diva.py
CHANGED
@@ -44,7 +44,7 @@ class WhisperConnector(nn.Module):
|
|
44 |
|
45 |
class DiVAModel(PreTrainedModel):
|
46 |
config_class = DiVAConfig
|
47 |
-
|
48 |
def __init__(
|
49 |
self, via_path=None, config_dict={}, device_map=None, speech_encoder_device=None
|
50 |
):
|
@@ -105,6 +105,10 @@ class DiVAModel(PreTrainedModel):
|
|
105 |
)
|
106 |
self.speech_encoder_device = speech_encoder_device
|
107 |
|
|
|
|
|
|
|
|
|
108 |
@classmethod
|
109 |
def from_pretrained(
|
110 |
cls,
|
|
|
44 |
|
45 |
class DiVAModel(PreTrainedModel):
|
46 |
config_class = DiVAConfig
|
47 |
+
|
48 |
def __init__(
|
49 |
self, via_path=None, config_dict={}, device_map=None, speech_encoder_device=None
|
50 |
):
|
|
|
105 |
)
|
106 |
self.speech_encoder_device = speech_encoder_device
|
107 |
|
108 |
+
|
109 |
+
def can_generate(cls):
|
110 |
+
return False
|
111 |
+
|
112 |
@classmethod
|
113 |
def from_pretrained(
|
114 |
cls,
|