Skip to content

Commit e84f22b

Browse files
aws-sadafaws-mesharma
authored andcommitted
Update n2-helper.py
1 parent 446ef6c commit e84f22b

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/helperscripts/n2-helper.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -670,6 +670,7 @@ def install_neuron_compiler_and_framework(self, args):
670670
str += '--pre '
671671
str += framework_name
672672
str += '==' + args.framework_version + '.*'
673+
str += ' mpmath==1.3'
673674
else:
674675
str += framework_name
675676
str += ' torchvision\n'

0 commit comments

Comments
 (0)