Skip to content

Commit

Permalink
fixing the inference build path when pre-building the inference op (#…
Browse files Browse the repository at this point in the history
  • Loading branch information
RezaYazdaniAminabadi authored Feb 11, 2022
1 parent 56eac38 commit d3cad05
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion op_builder/transformer_inference.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ def __init__(self, name=None):
super().__init__(name=name)

def absolute_name(self):
return f'deepspeed.ops.transformer_inference.{self.NAME}_op'
return f'deepspeed.ops.transformer.inference.{self.NAME}_op'

def sources(self):
return [
Expand Down

0 comments on commit d3cad05

Please sign in to comment.