Spaces:
Running
Running
add token to decoder models
Browse files- 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,
|