Upload pipeline.py with huggingface_hub
Browse files- pipeline.py +0 -1
pipeline.py
CHANGED
|
@@ -71,7 +71,6 @@ class StableDiffusionT5Pipeline(StableDiffusionPipeline):
|
|
| 71 |
# Ensure everything is properly registered for to("cuda")
|
| 72 |
self.register_modules(t5_projection=self.t5_projection)
|
| 73 |
|
| 74 |
-
@torch.no_grad()
|
| 75 |
def encode_prompt(
|
| 76 |
self,
|
| 77 |
prompt,
|
|
|
|
| 71 |
# Ensure everything is properly registered for to("cuda")
|
| 72 |
self.register_modules(t5_projection=self.t5_projection)
|
| 73 |
|
|
|
|
| 74 |
def encode_prompt(
|
| 75 |
self,
|
| 76 |
prompt,
|