Update app.py

app.py

@@ -269,6 +269,9 @@ def generate_30(
     # 3. Concatenate the text_encoder_2 embeddings 
     prompt_embeds2 = torch.cat([prompt_embeds_a2, prompt_embeds_b2])
     print('catted shape2: ', prompt_embeds2.shape)
+    prompt_embeds2 = torch.mean(prompt_embeds2,dim=0,keepdim=True)
+    print('averaged shape2: ', prompt_embeds.shape)
+
     pooled_prompt_embeds2 = torch.cat([pooled_prompt_embeds_a2, pooled_prompt_embeds_b2])
     print('catted pooled shape 2: ', pooled_prompt_embeds2.shape)
     pooled_prompt_embeds2 = torch.mean(pooled_prompt_embeds2,dim=0,keepdim=True)
@@ -276,6 +279,9 @@ def generate_30(
     pooled_prompt_embeds = torch.cat([pooled_prompt_embeds, pooled_prompt_embeds2],dim=1)
     print('catted combined meaned pooled shape: ', pooled_prompt_embeds.shape)
   
+    prompt_embeds = torch.cat([prompt_embeds, prompt_embeds2],dim=-1)
+    print('catted combined meaned shape: ', prompt_embeds.shape)
+
     options = {
         #"prompt": prompt,
         "prompt_embeds": prompt_embeds,
@@ -398,6 +404,9 @@ def generate_60(
     # 3. Concatenate the text_encoder_2 embeddings 
     prompt_embeds2 = torch.cat([prompt_embeds_a2, prompt_embeds_b2])
     print('catted shape2: ', prompt_embeds2.shape)
+    prompt_embeds2 = torch.mean(prompt_embeds2,dim=0,keepdim=True)
+    print('averaged shape2: ', prompt_embeds.shape)
+
     pooled_prompt_embeds2 = torch.cat([pooled_prompt_embeds_a2, pooled_prompt_embeds_b2])
     print('catted pooled shape 2: ', pooled_prompt_embeds2.shape)
     pooled_prompt_embeds2 = torch.mean(pooled_prompt_embeds2,dim=0,keepdim=True)
@@ -405,6 +414,9 @@ def generate_60(
     pooled_prompt_embeds = torch.cat([pooled_prompt_embeds, pooled_prompt_embeds2],dim=1)
     print('catted combined meaned pooled shape: ', pooled_prompt_embeds.shape)
   
+    prompt_embeds = torch.cat([prompt_embeds, prompt_embeds2],dim=-1)
+    print('catted combined meaned shape: ', prompt_embeds.shape)
+
     options = {
         #"prompt": prompt,
         "prompt_embeds": prompt_embeds,
@@ -527,6 +539,9 @@ def generate_90(
     # 3. Concatenate the text_encoder_2 embeddings 
     prompt_embeds2 = torch.cat([prompt_embeds_a2, prompt_embeds_b2])
     print('catted shape2: ', prompt_embeds2.shape)
+    prompt_embeds2 = torch.mean(prompt_embeds2,dim=0,keepdim=True)
+    print('averaged shape2: ', prompt_embeds.shape)
+
     pooled_prompt_embeds2 = torch.cat([pooled_prompt_embeds_a2, pooled_prompt_embeds_b2])
     print('catted pooled shape 2: ', pooled_prompt_embeds2.shape)
     pooled_prompt_embeds2 = torch.mean(pooled_prompt_embeds2,dim=0,keepdim=True)
@@ -534,6 +549,9 @@ def generate_90(
     pooled_prompt_embeds = torch.cat([pooled_prompt_embeds, pooled_prompt_embeds2],dim=1)
     print('catted combined meaned pooled shape: ', pooled_prompt_embeds.shape)
   
+    prompt_embeds = torch.cat([prompt_embeds, prompt_embeds2],dim=-1)
+    print('catted combined meaned shape: ', prompt_embeds.shape)
+
     options = {
         #"prompt": prompt,
         "prompt_embeds": prompt_embeds,