Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -13,7 +13,7 @@ from diffusers.utils import load_image
|
|
13 |
from PIL import Image
|
14 |
|
15 |
import torch
|
16 |
-
|
17 |
torch.backends.cuda.matmul.allow_tf32 = False
|
18 |
torch.backends.cuda.matmul.allow_bf16_reduced_precision_reduction = False
|
19 |
torch.backends.cuda.matmul.allow_fp16_reduced_precision_reduction = False
|
@@ -23,7 +23,7 @@ torch.backends.cudnn.benchmark = False
|
|
23 |
torch.backends.cuda.preferred_blas_library="cublas"
|
24 |
torch.backends.cuda.preferred_linalg_library="cusolver"
|
25 |
torch.set_float32_matmul_precision("highest")
|
26 |
-
|
27 |
os.putenv("HF_HUB_ENABLE_HF_TRANSFER","1")
|
28 |
os.environ["SAFETENSORS_FAST_GPU"] = "1"
|
29 |
os.putenv("TOKENIZERS_PARALLELISM","False")
|
|
|
13 |
from PIL import Image
|
14 |
|
15 |
import torch
|
16 |
+
|
17 |
torch.backends.cuda.matmul.allow_tf32 = False
|
18 |
torch.backends.cuda.matmul.allow_bf16_reduced_precision_reduction = False
|
19 |
torch.backends.cuda.matmul.allow_fp16_reduced_precision_reduction = False
|
|
|
23 |
torch.backends.cuda.preferred_blas_library="cublas"
|
24 |
torch.backends.cuda.preferred_linalg_library="cusolver"
|
25 |
torch.set_float32_matmul_precision("highest")
|
26 |
+
|
27 |
os.putenv("HF_HUB_ENABLE_HF_TRANSFER","1")
|
28 |
os.environ["SAFETENSORS_FAST_GPU"] = "1"
|
29 |
os.putenv("TOKENIZERS_PARALLELISM","False")
|