change shape

app.py

@@ -48,7 +48,7 @@ if device=="cpu":
     # pipeline_t2i.safety_checker = pipeline_t2i.safety_checker.to(memory_format=torch.channels_last)
 
     # Create random input to enable JIT compilation
-    sample = torch.randn(2,4,64,64).type(torch_dtype)
+    sample = torch.randn(2,4,64,48).type(torch_dtype)
     timestep = torch.rand(1)*999
     encoder_hidden_status = torch.randn(2,77,768).type(torch_dtype)
     input_example = (sample, timestep, encoder_hidden_status)