move_model_to_devices no longer crashes if you don't have accelerate

2022-06-21 13:15:46 -04:00 · 2022-06-21 13:15:46 -04:00 · 6b172306f6
parent f2c5bb5cb7
commit 6b172306f6
1 changed files with 2 additions and 1 deletions
--- a/aiserver.py
+++ b/aiserver.py
@ -633,8 +633,9 @@ def move_model_to_devices(model):
        generator = model.generate
        return

+    import breakmodel
+
    if(utils.HAS_ACCELERATE):
-        import breakmodel
        disk_blocks = breakmodel.disk_blocks
        gpu_blocks = breakmodel.gpu_blocks
        ram_blocks = len(utils.layers_module_names) - sum(gpu_blocks)