Spaces:
Running
on
Zero
Running
on
Zero
1inkusFace
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -34,7 +34,7 @@ torch.backends.cudnn.benchmark = False
|
|
34 |
#torch.backends.cuda.preferred_linalg_library="cusolver"
|
35 |
|
36 |
hftoken = os.getenv("HF_AUTH_TOKEN")
|
37 |
-
|
38 |
|
39 |
code = r'''
|
40 |
import torch
|
|
|
34 |
#torch.backends.cuda.preferred_linalg_library="cusolver"
|
35 |
|
36 |
hftoken = os.getenv("HF_AUTH_TOKEN")
|
37 |
+
os.environ["SAFETENSORS_FAST_GPU"] = "1"
|
38 |
|
39 |
code = r'''
|
40 |
import torch
|