Videobook_v1.5

Runtime error

Warlord-K commited on Jul 4, 2023

Commit

28d3917

•

1 Parent(s): f118d8b

Fix Bug

Files changed (1) hide show

Videobook/Videobook.py CHANGED Viewed

@@ -29,7 +29,7 @@ class Videobook:
   def generate_imgs(self, sentences, steps):
     imgs = []
     for sentence in sentences:
-      sentence = self.style + ' of ' + sentence + ', ' + self.tags
       imgs.append(self.pipe.generate(prompt = sentence['pos'], negative_prompt = sentence['neg'], num_inference_steps = steps))
     return imgs

   def generate_imgs(self, sentences, steps):
     imgs = []
     for sentence in sentences:
+      sentence['pos'] = self.style + ' of ' + sentence['pos'] + ', ' + self.tags
       imgs.append(self.pipe.generate(prompt = sentence['pos'], negative_prompt = sentence['neg'], num_inference_steps = steps))
     return imgs