add try-except to replicate

cog.yaml

@@ -1,5 +1,5 @@
 build:
-  cuda: "11.5.1"
+  cuda: "11.4"
   gpu: true
   python_version: "3.10"
   system_packages:

replicate_predictor.py

@@ -31,6 +31,7 @@ class ReplicatePredictor(BasePredictor):
             default=4
         ),
     ) -> Iterator[Path]:
+        try: 
             seed = -1
             log2_mid_count = 3 if intermediate_outputs else 0
             image_stream = self.model.generate_image_stream(
@@ -49,3 +50,8 @@ class ReplicatePredictor(BasePredictor):
                 image_path = path / 'min-dalle-iter-{}.jpg'.format(iter)
                 image.save(str(image_path))
                 yield image_path
+        except:
+            print("An error occured, deleting model")
+            del self.model
+            self.setup()
+            raise Exception("There was an error, please try again")