rodrigomasini commited on
Commit
dbcbd21
1 Parent(s): 63cb966

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -93,6 +93,7 @@ import diffusers
93
  from diffusers import StableDiffusionXLPipeline
94
  from diffusers import DDIMScheduler
95
  import torch.nn.functional as F
 
96
 
97
  def cal_attn_mask(total_length,id_length,sa16,sa32,sa64,device="cuda",dtype= torch.float16):
98
  bool_matrix256 = torch.rand((1, total_length * 256),device = device,dtype = dtype) < sa16
 
93
  from diffusers import StableDiffusionXLPipeline
94
  from diffusers import DDIMScheduler
95
  import torch.nn.functional as F
96
+ from transformers.models.clip.configuration_clip import CLIPVisionConfig
97
 
98
  def cal_attn_mask(total_length,id_length,sa16,sa32,sa64,device="cuda",dtype= torch.float16):
99
  bool_matrix256 = torch.rand((1, total_length * 256),device = device,dtype = dtype) < sa16