revert muh clean code

2023-03-10 00:56:29 +07:00 · 2023-03-10 00:56:29 +07:00 · fec0685405
parent 0514f011ff
commit fec0685405
1 changed files with 8 additions and 0 deletions
--- a/tortoise/utils/device.py
+++ b/tortoise/utils/device.py
@ -61,9 +61,17 @@ def get_device_vram( name=get_device_name() ):
 def get_device_batch_size(name=None):
    vram = get_device_vram(name)

+    if vram > 14:
+        return 16
+    elif vram > 10:
+        return 8
+    elif vram > 7:
+        return 4
+    """
    for k, v in DEVICE_BATCH_SIZE_MAP:
        if vram > k:
            return v
+    """
    return 1

 def get_device_count(name=get_device_name()):