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

add token to decoder models

Browse files
Files changed (1) hide show
  1. optimum_neuron_export.py +1 -1
optimum_neuron_export.py CHANGED
@@ -325,7 +325,7 @@ def export_decoder_model(model_id: str, folder: str, token:str) -> Generator:
325
 
326
  try:
327
  # Trigger the export/compilation
328
- neuron_config = NeuronModelForCausalLM.get_neuron_config(model_name_or_path=model_id, **export_kwargs)
329
  neuron_config.target = "inf2"
330
  model = NeuronModelForCausalLM.export(
331
  model_id=model_id, neuron_config=neuron_config,
 
325
 
326
  try:
327
  # Trigger the export/compilation
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,