fix error and pull in upstream changes

truib · truib · commit ec5bec5ee656 · 2025-05-23T12:46:12.000+02:00
diff --git a/eb_hooks.py b/eb_hooks.py
@@ -1252,7 +1252,7 @@ def post_module_hook(self, *args, **kwargs):
 }
 
 POST_MODULE_HOOKS = {}
-Torch
+
 # Define parallelism limit operations
 def divide_by_factor(parallel, factor):
     """Divide parallelism by given factor"""

Original file line number	Diff line number	Diff line change
`@@ -1252,7 +1252,7 @@ def post_module_hook(self, args, *kwargs):`
`1252`	`1252`	`}`
`1253`	`1253`
`1254`	`1254`	`POST_MODULE_HOOKS = {}`
`1255`		`-Torch`
	`1255`	`+`
`1256`	`1256`	`# Define parallelism limit operations`
`1257`	`1257`	`def divide_by_factor(parallel, factor):`
`1258`	`1258`	`"""Divide parallelism by given factor"""`