We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 7dfe757 commit d04db8dCopy full SHA for d04db8d
fastsafetensors/loader.py
@@ -60,8 +60,10 @@ def __init__(
60
global loaded_nvidia
61
if not loaded_nvidia:
62
fstcpp.load_nvidia_functions()
63
- if fstcpp.init_gds() != 0:
64
- raise Exception(f"[FAIL] init_gds()")
+ if not nogds:
+ # no need to init gds and consume 10s+ in none-gds case
65
+ if fstcpp.init_gds() != 0:
66
+ raise Exception(f"[FAIL] init_gds()")
67
loaded_nvidia = True
68
global gl_set_numa
69
if not gl_set_numa and set_numa:
0 commit comments