Skip to content

Commit

Permalink
fix format
Browse files Browse the repository at this point in the history
  • Loading branch information
echarlaix committed Mar 6, 2024
1 parent 600bd78 commit 9536181
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions src/lighteval/models/optimum_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,10 +10,7 @@


class OptimumModel(BaseModel):


def _create_auto_model(self, config: OptimumModelConfig, env_config: EnvConfig):

# TODO : Get loading class from optimum config (add support for ORTModelForCausalLM / INCModelForCausalLM / IPEXModelForCausalLM)
# optimum_config = OptimumConfig.from_pretrained(config.pretrained)

Expand Down

0 comments on commit 9536181

Please sign in to comment.