Update gradio_utils.py

gradio_utils.py

@@ -42,7 +42,6 @@ class SpatialAttnProcessor2_0(torch.nn.Module):
         temb=None):
         # un_cond_hidden_states, cond_hidden_states = hidden_states.chunk(2)
         # un_cond_hidden_states = self.__call2__(attn, un_cond_hidden_states,encoder_hidden_states,attention_mask,temb)
-        # 生成一个0到1之间的随机数
         global total_count,attn_count,cur_step,mask256,mask1024,mask4096
         global sa16, sa32, sa64
         global write
@@ -50,7 +49,6 @@ class SpatialAttnProcessor2_0(torch.nn.Module):
             self.id_bank[cur_step] = [hidden_states[:self.id_length], hidden_states[self.id_length:]]
         else:
             encoder_hidden_states = torch.cat(self.id_bank[cur_step][0],hidden_states[:1],self.id_bank[cur_step][1],hidden_states[1:])
-        # 判断随机数是否大于0.5
         if cur_step <5:
             hidden_states = self.__call2__(attn, hidden_states,encoder_hidden_states,attention_mask,temb)
         else:   # 256 1024 4096
@@ -260,7 +258,6 @@ def cal_attn_indice_xl_effcient_memory(total_length,id_length,sa32,sa64,height,w
     nums_4096 = (height // 16) * (width // 16)
     bool_matrix1024 = torch.rand((total_length,nums_1024),device = device,dtype = dtype) < sa32
     bool_matrix4096 = torch.rand((total_length,nums_4096),device = device,dtype = dtype) < sa64
-    # 用nonzero()函数获取所有为True的值的索引
     indices1024 = [torch.nonzero(bool_matrix1024[i], as_tuple=True)[0] for i in range(total_length)]
     indices4096 = [torch.nonzero(bool_matrix4096[i], as_tuple=True)[0] for i in range(total_length)]
 
@@ -431,7 +428,6 @@ def is_torch2_available():
     return hasattr(F, "scaled_dot_product_attention")
 
 
-# 将列表转换为字典的函数
 def character_to_dict(general_prompt):
     character_dict = {}    
     generate_prompt_arr = general_prompt.splitlines()
@@ -439,7 +435,6 @@ def character_to_dict(general_prompt):
     invert_character_index_dict = {}
     character_list = []
     for ind,string in enumerate(generate_prompt_arr):
-        # 分割字符串寻找key和value
         start = string.find('[')
         end = string.find(']')
         if start != -1 and end != -1: