Update file export.py
Browse files
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 |
-
|
| 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)
|