Skip to content

Commit

Permalink
Update accelerator
Browse files Browse the repository at this point in the history
  • Loading branch information
liuhongwei committed May 27, 2024
1 parent 8fe4ceb commit a4bac18
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion opencompass/utils/run.py
Original file line number Diff line number Diff line change
Expand Up @@ -279,7 +279,8 @@ def change_accelerator(models, accelerator):
else:
raise ValueError(f'Unsupported accelerator {accelerator} for model type {model["type"]}')
else:
raise ValueError(f'Unsupported model type {model["type"]}')
acc_model = model
logger.warning(f'Unsupported model type {model["type"]}, will keep the original model.')
model_accels.append(acc_model)
return model_accels

Expand Down

0 comments on commit a4bac18

Please sign in to comment.