Update app.py

app.py

@@ -104,7 +104,13 @@ def merge_and_run(prompt, negative_prompt, shuffled_items, lora_1_scale=0.5, lor
   #pipe = copy.deepcopy(original_pipe)
   start_time = time()
   unet = copy.deepcopy(original_pipe.unet)
-  pipe = StableDiffusionXLPipeline(vae=original_pipe.vae, text_encoder=original_pipe.text_encoder, text_encoder_2=original_pipe.text_encoder_2, scheduler=original_pipe.scheduler, unet=unet)
+  pipe = StableDiffusionXLPipeline(vae=original_pipe.vae,
+                                   text_encoder=original_pipe.text_encoder,
+                                   text_encoder_2=original_pipe.text_encoder_2,
+                                   scheduler=original_pipe.scheduler,
+                                   tokenizer=original_pipe.tokenizer,
+                                   tokenizer_2=original_pipe.tokenizer_2,
+                                   unet=unet)
   pickle_time = time() - start_time
   print(f"copy time: {pickle_time}")  
   pipe.to("cuda")