flpelerin commited on
Commit
dba3aeb
·
1 Parent(s): 4a94319

Update file export.py

Browse files
Files changed (1) hide show
  1. export.py +1 -1
export.py CHANGED
@@ -131,7 +131,7 @@ def ExportModelFP32(model, config, output_path):
131
  WriteWeightsFP32(out_file, model, 'embedding.weight')
132
 
133
  WriteLayersFP32(out_file, model, 'layers.%d.mixer.in_proj.weight', n_layers)
134
- writeLayersFP32(out_file, model, 'layers.%d.mixer.conv1d.weight', n_layers)
135
  WriteLayersFP32(out_file, model, 'layers.%d.mixer.conv1d.bias', n_layers)
136
  WriteLayersFP32(out_file, model, 'layers.%d.mixer.x_proj.weight', n_layers)
137
  WriteLayersFP32(out_file, model, 'layers.%d.mixer.dt_proj.weight', n_layers)
 
131
  WriteWeightsFP32(out_file, model, 'embedding.weight')
132
 
133
  WriteLayersFP32(out_file, model, 'layers.%d.mixer.in_proj.weight', n_layers)
134
+ WriteLayersFP32(out_file, model, 'layers.%d.mixer.conv1d.weight', n_layers)
135
  WriteLayersFP32(out_file, model, 'layers.%d.mixer.conv1d.bias', n_layers)
136
  WriteLayersFP32(out_file, model, 'layers.%d.mixer.x_proj.weight', n_layers)
137
  WriteLayersFP32(out_file, model, 'layers.%d.mixer.dt_proj.weight', n_layers)