Update model.py

model.py

@@ -1128,7 +1128,7 @@ class StreamMultiDiffusion(nn.Module):
                 encoder_hidden_states=self.prompt_embeds,  # (B, 77, 768)
                 return_dict=False,
             )[0]  # (B, 4, h, w)
-        except e:
+        except Exception as e:
             print(e)
         print('222222222222222', model_pred.dtype)