Pix2Pix-Video

Runtime error

fffiloni commited on Apr 12, 2023

Commit

3a96db7

1 Parent(s): 822f411

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,8 +15,8 @@ import random
 pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained("timbrooks/instruct-pix2pix", torch_dtype=torch.float16, safety_checker=None)
 pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
-#pipe.enable_xformers_memory_efficient_attention()
-#pipe.unet.to(memory_format=torch.channels_last)
 device = "GPU 🔥" if torch.cuda.is_available() else "CPU 🥶"

 pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained("timbrooks/instruct-pix2pix", torch_dtype=torch.float16, safety_checker=None)
 pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
+pipe.enable_xformers_memory_efficient_attention()
+pipe.unet.to(memory_format=torch.channels_last)
 device = "GPU 🔥" if torch.cuda.is_available() else "CPU 🥶"