ford442 commited on
Commit
3bbe93c
·
verified ·
1 Parent(s): 9680d6b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ import paramiko
20
  import datetime
21
  #from diffusers import DPMSolverSDEScheduler
22
  from diffusers.models.attention_processor import AttnProcessor2_0
23
-
24
  torch.backends.cuda.matmul.allow_tf32 = False
25
  torch.backends.cuda.matmul.allow_bf16_reduced_precision_reduction = False
26
  torch.backends.cuda.matmul.allow_fp16_reduced_precision_reduction = False
 
20
  import datetime
21
  #from diffusers import DPMSolverSDEScheduler
22
  from diffusers.models.attention_processor import AttnProcessor2_0
23
+ from transformers import CLIPTextModelWithProjection, CLIPTextModel
24
  torch.backends.cuda.matmul.allow_tf32 = False
25
  torch.backends.cuda.matmul.allow_bf16_reduced_precision_reduction = False
26
  torch.backends.cuda.matmul.allow_fp16_reduced_precision_reduction = False