fal 1.13.1__py3-none-any.whl → 1.13.3__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 CHANGED
@@ -17,5 +17,5 @@ __version__: str
17
17
  __version_tuple__: VERSION_TUPLE
18
18
  version_tuple: VERSION_TUPLE
19
19
 
20
- __version__ = version = '1.13.1'
21
- __version_tuple__ = version_tuple = (1, 13, 1)
20
+ __version__ = version = '1.13.3'
21
+ __version_tuple__ = version_tuple = (1, 13, 3)
fal/exceptions/_cuda.py CHANGED
@@ -26,12 +26,22 @@ 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
- return (
30
- isinstance(exception, RuntimeError)
31
- and len(exception.args) == 1
32
- and "CUDA" in exception.args[0]
33
- and "out of memory" in exception.args[0]
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
+ if "CUDNN_STATUS_INTERNAL_ERROR" in msg:
42
+ return True
43
+
44
+ return False
35
45
 
36
46
 
37
47
  # based on https://github.com/BlackHC/toma/blob/master/toma/torch_cuda_memory.py
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: fal
3
- Version: 1.13.1
3
+ Version: 1.13.3
4
4
  Summary: fal is an easy-to-use Serverless Python Framework
5
5
  Author: Features & Labels <support@fal.ai>
6
6
  Requires-Python: >=3.8
@@ -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=Ac1kTVbQ6uA52UVlXW9_b4jmcDcK6jwecj43M7ul0Qw,513
3
+ fal/_fal_version.py,sha256=GwK4kZgtx2CVfLLI6BiywAgeTZLJmNUqgVGObzf1JRA,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=q5EPFYEb7Iyw03cHrQlRHnH5xOvjwTwQdM6a9N3GB8k,1494
45
+ fal/exceptions/_cuda.py,sha256=L3qvDNaPTthp95IFSBI6pMt3YbRfn1H0inQkj_7NKF8,1719
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.1.dist-info/METADATA,sha256=wQj0ZUks0AemEmsAw1HEc82p5h0mmRDupMVArnAs07w,4062
138
- fal-1.13.1.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
139
- fal-1.13.1.dist-info/entry_points.txt,sha256=32zwTUC1U1E7nSTIGCoANQOQ3I7-qHG5wI6gsVz5pNU,37
140
- fal-1.13.1.dist-info/top_level.txt,sha256=r257X1L57oJL8_lM0tRrfGuXFwm66i1huwQygbpLmHw,21
141
- fal-1.13.1.dist-info/RECORD,,
137
+ fal-1.13.3.dist-info/METADATA,sha256=utyInTbohqq3mhB6vu-QJ_57UBFK55w-yiOJ-MOJVX0,4062
138
+ fal-1.13.3.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
139
+ fal-1.13.3.dist-info/entry_points.txt,sha256=32zwTUC1U1E7nSTIGCoANQOQ3I7-qHG5wI6gsVz5pNU,37
140
+ fal-1.13.3.dist-info/top_level.txt,sha256=r257X1L57oJL8_lM0tRrfGuXFwm66i1huwQygbpLmHw,21
141
+ fal-1.13.3.dist-info/RECORD,,
@@ -1,5 +1,5 @@
1
1
  Wheel-Version: 1.0
2
- Generator: setuptools (78.1.0)
2
+ Generator: setuptools (79.0.1)
3
3
  Root-Is-Purelib: true
4
4
  Tag: py3-none-any
5
5