Spaces:

JMalott
/

ai_architecture

Runtime error

JMalott commited on Sep 6, 2022

Commit

f9a4923

1 Parent(s): 42e69b2

Update min_dalle/min_dalle.py

Files changed (1) hide show

min_dalle/min_dalle.py CHANGED Viewed

@@ -257,6 +257,7 @@ class MinDalle:
                     is_seamless=is_seamless,
                     is_verbose=is_verbose
                 )
     def generate_image_stream(self, *args, **kwargs) -> Iterator[Image.Image]:
         image_stream = self.generate_raw_image_stream(*args, **kwargs)

                     is_seamless=is_seamless,
                     is_verbose=is_verbose
                 )
+            del attention_state
     def generate_image_stream(self, *args, **kwargs) -> Iterator[Image.Image]:
         image_stream = self.generate_raw_image_stream(*args, **kwargs)