Spaces:
Running
Running
Update optimum_neuron_export.py
Browse files- optimum_neuron_export.py +1 -1
optimum_neuron_export.py
CHANGED
|
@@ -328,7 +328,7 @@ def export_decoder_model(model_id: str, folder: str, token:str) -> Generator:
|
|
| 328 |
neuron_config = NeuronModelForCausalLM.get_neuron_config(model_name_or_path=model_id, token=token, **export_kwargs)
|
| 329 |
neuron_config.target = "inf2"
|
| 330 |
model = NeuronModelForCausalLM.export(
|
| 331 |
-
model_id=model_id, neuron_config=neuron_config,
|
| 332 |
)
|
| 333 |
model.save_pretrained(folder)
|
| 334 |
|
|
|
|
| 328 |
neuron_config = NeuronModelForCausalLM.get_neuron_config(model_name_or_path=model_id, token=token, **export_kwargs)
|
| 329 |
neuron_config.target = "inf2"
|
| 330 |
model = NeuronModelForCausalLM.export(
|
| 331 |
+
model_id=model_id, neuron_config=neuron_config, token=token,
|
| 332 |
)
|
| 333 |
model.save_pretrained(folder)
|
| 334 |
|