Feature Extraction
Transformers
Safetensors
diva
custom_code
WillHeld commited on
Commit
f79cc7c
·
verified ·
1 Parent(s): 27d35c6

Update modeling_diva.py

Browse files
Files changed (1) hide show
  1. 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,