fabric

Runtime error

dvruette commited on Jul 20, 2023

Commit

50911e6

•

1 Parent(s): 44e4d0e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -20,6 +20,8 @@ class GeneratorWrapper:
     self.reload()
   def generate(self, *args, **kwargs):
     return self.generator.generate(*args, **kwargs)
   def to(self, device):

     self.reload()
   def generate(self, *args, **kwargs):
+    if not hasattr(self, "generator"):
+        self.reload()
     return self.generator.generate(*args, **kwargs)
   def to(self, device):