fal 1.13.1__py3-none-any.whl → 1.13.2__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of fal might be problematic. Click here for more details.
- fal/_fal_version.py +2 -2
- fal/exceptions/_cuda.py +13 -6
- {fal-1.13.1.dist-info → fal-1.13.2.dist-info}/METADATA +1 -1
- {fal-1.13.1.dist-info → fal-1.13.2.dist-info}/RECORD +7 -7
- {fal-1.13.1.dist-info → fal-1.13.2.dist-info}/WHEEL +1 -1
- {fal-1.13.1.dist-info → fal-1.13.2.dist-info}/entry_points.txt +0 -0
- {fal-1.13.1.dist-info → fal-1.13.2.dist-info}/top_level.txt +0 -0
fal/_fal_version.py
CHANGED
fal/exceptions/_cuda.py
CHANGED
|
@@ -26,12 +26,19 @@ def _is_cuda_oom_exception(exception: BaseException) -> bool:
|
|
|
26
26
|
|
|
27
27
|
# based on https://github.com/BlackHC/toma/blob/master/toma/torch_cuda_memory.py
|
|
28
28
|
def _is_cuda_out_of_memory(exception: BaseException) -> bool:
|
|
29
|
-
|
|
30
|
-
|
|
31
|
-
|
|
32
|
-
|
|
33
|
-
|
|
34
|
-
|
|
29
|
+
if not isinstance(exception, RuntimeError) or len(exception.args) != 1:
|
|
30
|
+
return False
|
|
31
|
+
|
|
32
|
+
msg = exception.args[0]
|
|
33
|
+
|
|
34
|
+
if "CUDA" in msg and "out of memory" in msg:
|
|
35
|
+
return True
|
|
36
|
+
|
|
37
|
+
# https://github.com/pytorch/pytorch/issues/112377
|
|
38
|
+
if "NVML_SUCCESS == r INTERNAL ASSERT FAILED" in msg:
|
|
39
|
+
return True
|
|
40
|
+
|
|
41
|
+
return False
|
|
35
42
|
|
|
36
43
|
|
|
37
44
|
# based on https://github.com/BlackHC/toma/blob/master/toma/torch_cuda_memory.py
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
fal/__init__.py,sha256=wXs1G0gSc7ZK60-bHe-B2m0l_sA6TrFk4BxY0tMoLe8,784
|
|
2
2
|
fal/__main__.py,sha256=4JMK66Wj4uLZTKbF-sT3LAxOsr6buig77PmOkJCRRxw,83
|
|
3
|
-
fal/_fal_version.py,sha256=
|
|
3
|
+
fal/_fal_version.py,sha256=hTrFYarxhkgFoSWsNux9T4qa4NvhnWLOscYReFS8bQs,513
|
|
4
4
|
fal/_serialization.py,sha256=rD2YiSa8iuzCaZohZwN_MPEB-PpSKbWRDeaIDpTEjyY,7653
|
|
5
5
|
fal/_version.py,sha256=EBGqrknaf1WygENX-H4fBefLvHryvJBBGtVJetaB0NY,266
|
|
6
6
|
fal/api.py,sha256=dNl6c_CvoFqG5tGXcA6nH6zhuHUDQY3KKLZfNYAIGGo,45400
|
|
@@ -42,7 +42,7 @@ fal/console/icons.py,sha256=De9MfFaSkO2Lqfne13n3PrYfTXJVIzYZVqYn5BWsdrA,108
|
|
|
42
42
|
fal/console/ux.py,sha256=KMQs3UHQvVHDxDQQqlot-WskVKoMQXOE3jiVkkfmIMY,356
|
|
43
43
|
fal/exceptions/__init__.py,sha256=m2okJEpax11mnwmoqO_pCGtbt-FvzKiiuMhKo2ok-_8,270
|
|
44
44
|
fal/exceptions/_base.py,sha256=LwzpMaW_eYQEC5s26h2qGXbNA-S4bOqC8s-bMCX6HjE,1491
|
|
45
|
-
fal/exceptions/_cuda.py,sha256=
|
|
45
|
+
fal/exceptions/_cuda.py,sha256=bajgVfzubYjxJIwFlvUyBhPobX8HnZozCANUDconV94,1653
|
|
46
46
|
fal/exceptions/auth.py,sha256=gxRago5coI__vSIcdcsqhhq1lRPkvCnwPAueIaXTAdw,329
|
|
47
47
|
fal/logging/__init__.py,sha256=avgKA2V8GJeUtZuWZJjSYgkkrXKpuDMS-YiIBmLda7w,1599
|
|
48
48
|
fal/logging/isolate.py,sha256=jIryi46ZVlJ1mfan4HLNQQ3jwMi8z-WwfqqLlttQVkc,2449
|
|
@@ -134,8 +134,8 @@ openapi_fal_rest/models/workflow_node_type.py,sha256=-FzyeY2bxcNmizKbJI8joG7byRi
|
|
|
134
134
|
openapi_fal_rest/models/workflow_schema.py,sha256=4K5gsv9u9pxx2ItkffoyHeNjBBYf6ur5bN4m_zePZNY,2019
|
|
135
135
|
openapi_fal_rest/models/workflow_schema_input.py,sha256=2OkOXWHTNsCXHWS6EGDFzcJKkW5FIap-2gfO233EvZQ,1191
|
|
136
136
|
openapi_fal_rest/models/workflow_schema_output.py,sha256=EblwSPAGfWfYVWw_WSSaBzQVju296is9o28rMBAd0mc,1196
|
|
137
|
-
fal-1.13.
|
|
138
|
-
fal-1.13.
|
|
139
|
-
fal-1.13.
|
|
140
|
-
fal-1.13.
|
|
141
|
-
fal-1.13.
|
|
137
|
+
fal-1.13.2.dist-info/METADATA,sha256=WYWB2pvBi7a0iqR_lfx9aKKSUIwoUG4F_SX5th7POsY,4062
|
|
138
|
+
fal-1.13.2.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
|
|
139
|
+
fal-1.13.2.dist-info/entry_points.txt,sha256=32zwTUC1U1E7nSTIGCoANQOQ3I7-qHG5wI6gsVz5pNU,37
|
|
140
|
+
fal-1.13.2.dist-info/top_level.txt,sha256=r257X1L57oJL8_lM0tRrfGuXFwm66i1huwQygbpLmHw,21
|
|
141
|
+
fal-1.13.2.dist-info/RECORD,,
|
|
File without changes
|
|
File without changes
|