resolve 3060

huggingface · Nov 22, 2024 · 9a9b0fa · 9a9b0fa
1 parent 08101b9
commit 9a9b0fa
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/accelerate/hooks.py b/src/accelerate/hooks.py
@@ -436,7 +436,8 @@ def attach_execution_device_hook(
         return
 
     for child in module.children():
-        attach_execution_device_hook(child, execution_device, skip_keys=skip_keys, tied_params_map=tied_params_map)
+        attach_execution_device_hook(child, execution_device, skip_keys=skip_keys, 
+            preload_module_classes=preload_module_classes, tied_params_map=tied_params_map)
 
 
 def attach_align_device_hook(