1inkusFace commited on
Commit
e733a7b
·
verified ·
1 Parent(s): 04cedad

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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")