RealVis_v5.0_BF16_F

Runtime error

1inkusFace commited on Jan 22

Commit

c8fa40a

verified ·

1 Parent(s): 7e206d8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -273,7 +273,7 @@ def generate_30(
     print('catted pooled shape 2: ', pooled_prompt_embeds2.shape)
     pooled_prompt_embeds2 = torch.mean(pooled_prompt_embeds2,dim=1)
     print('pooled meaned shape 2: ', pooled_prompt_embeds2.shape)
-    pooled_prompt_embeds = torch.cat([pooled_prompt_embeds, pooled_prompt_embeds2])
     print('catted combined meaned pooled shape: ', pooled_prompt_embeds.shape)
     options = {
@@ -402,7 +402,7 @@ def generate_60(
     print('catted pooled shape 2: ', pooled_prompt_embeds2.shape)
     pooled_prompt_embeds2 = torch.mean(pooled_prompt_embeds2,dim=1)
     print('pooled meaned shape 2: ', pooled_prompt_embeds2.shape)
-    pooled_prompt_embeds = torch.cat([pooled_prompt_embeds, pooled_prompt_embeds2])
     print('catted combined meaned pooled shape: ', pooled_prompt_embeds.shape)
     options = {
@@ -531,7 +531,7 @@ def generate_90(
     print('catted pooled shape 2: ', pooled_prompt_embeds2.shape)
     pooled_prompt_embeds2 = torch.mean(pooled_prompt_embeds2,dim=1)
     print('pooled meaned shape 2: ', pooled_prompt_embeds2.shape)
-    pooled_prompt_embeds = torch.cat([pooled_prompt_embeds, pooled_prompt_embeds2])
     print('catted combined meaned pooled shape: ', pooled_prompt_embeds.shape)
     options = {

     print('catted pooled shape 2: ', pooled_prompt_embeds2.shape)
     pooled_prompt_embeds2 = torch.mean(pooled_prompt_embeds2,dim=1)
     print('pooled meaned shape 2: ', pooled_prompt_embeds2.shape)
+    pooled_prompt_embeds = torch.cat([pooled_prompt_embeds, pooled_prompt_embeds2],dim=2)
     print('catted combined meaned pooled shape: ', pooled_prompt_embeds.shape)
     options = {
     print('catted pooled shape 2: ', pooled_prompt_embeds2.shape)
     pooled_prompt_embeds2 = torch.mean(pooled_prompt_embeds2,dim=1)
     print('pooled meaned shape 2: ', pooled_prompt_embeds2.shape)
+    pooled_prompt_embeds = torch.cat([pooled_prompt_embeds, pooled_prompt_embeds2],dim=2)
     print('catted combined meaned pooled shape: ', pooled_prompt_embeds.shape)
     options = {
     print('catted pooled shape 2: ', pooled_prompt_embeds2.shape)
     pooled_prompt_embeds2 = torch.mean(pooled_prompt_embeds2,dim=1)
     print('pooled meaned shape 2: ', pooled_prompt_embeds2.shape)
+    pooled_prompt_embeds = torch.cat([pooled_prompt_embeds, pooled_prompt_embeds2],dim=2)
     print('catted combined meaned pooled shape: ', pooled_prompt_embeds.shape)
     options = {