opendiffusionai
/

stablediffusion_t5

ppbrown commited on May 31

Commit

e89ef5c

verified ·

1 Parent(s): a2aa4b5

Upload pipeline.py with huggingface_hub

Files changed (1) hide show

pipeline.py CHANGED Viewed

@@ -71,7 +71,6 @@ class StableDiffusionT5Pipeline(StableDiffusionPipeline):
         # Ensure everything is properly registered for to("cuda")
         self.register_modules(t5_projection=self.t5_projection)
-    @torch.no_grad()
     def encode_prompt(
         self,
         prompt,

         # Ensure everything is properly registered for to("cuda")
         self.register_modules(t5_projection=self.t5_projection)
     def encode_prompt(
         self,
         prompt,