badaoui HF Staff commited on
Commit
50ffb54
·
verified ·
1 Parent(s): a280be9

Update optimum_neuron_export.py

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