1inkusFace commited on
Commit
8f319fe
·
verified ·
1 Parent(s): 4d756fd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -34,12 +34,12 @@ from image_gen_aux import UpscaleWithModel
34
  #from diffusers.models.attention_processor import AttnProcessor2_0
35
  from transformers import CLIPTextModelWithProjection, CLIPTextModel, CLIPTokenizer
36
 
37
- torch.backends.cuda.matmul.allow_tf32 = False
38
  torch.backends.cuda.matmul.allow_bf16_reduced_precision_reduction = False
39
  torch.backends.cuda.matmul.allow_fp16_reduced_precision_reduction = False
40
- torch.backends.cudnn.allow_tf32 = False
41
  torch.backends.cudnn.deterministic = False
42
- torch.backends.cudnn.benchmark = False
43
  torch.backends.cuda.preferred_blas_library="cublas"
44
  torch.backends.cuda.preferred_linalg_library="cusolver"
45
 
@@ -118,8 +118,7 @@ model_repo='John6666/uber-realistic-porn-merge-xl-urpmxl-v6final-sdxl'
118
  #'John6666/pornworks-real-porn-ponyv04-sdxl'
119
  #'John6666/pornmaster-amateur-sdxlv1vae-sdxl'
120
 
121
-
122
- rv='ford442/RealVisXL_V5.0_BF16'
123
 
124
  text_encoder = CLIPTextModel.from_pretrained(rv, subfolder='text_encoder', token=True)#.to(device=device, dtype=torch.bfloat16)
125
  text_encoder_2 = CLIPTextModelWithProjection.from_pretrained(model_repo, subfolder='text_encoder_2',token=True)#.to(device=device, dtype=torch.bfloat16)
 
34
  #from diffusers.models.attention_processor import AttnProcessor2_0
35
  from transformers import CLIPTextModelWithProjection, CLIPTextModel, CLIPTokenizer
36
 
37
+ torch.backends.cuda.matmul.allow_tf32 = True
38
  torch.backends.cuda.matmul.allow_bf16_reduced_precision_reduction = False
39
  torch.backends.cuda.matmul.allow_fp16_reduced_precision_reduction = False
40
+ torch.backends.cudnn.allow_tf32 = True
41
  torch.backends.cudnn.deterministic = False
42
+ torch.backends.cudnn.benchmark = True
43
  torch.backends.cuda.preferred_blas_library="cublas"
44
  torch.backends.cuda.preferred_linalg_library="cusolver"
45
 
 
118
  #'John6666/pornworks-real-porn-ponyv04-sdxl'
119
  #'John6666/pornmaster-amateur-sdxlv1vae-sdxl'
120
 
121
+ rv='ford442/RealVisXL_V5.0_BF64'
 
122
 
123
  text_encoder = CLIPTextModel.from_pretrained(rv, subfolder='text_encoder', token=True)#.to(device=device, dtype=torch.bfloat16)
124
  text_encoder_2 = CLIPTextModelWithProjection.from_pretrained(model_repo, subfolder='text_encoder_2',token=True)#.to(device=device, dtype=torch.bfloat16)