Commit
·
c9da6df
1
Parent(s):
54c25b9
Update pipeline.py
Browse files- pipeline.py +1 -1
pipeline.py
CHANGED
|
@@ -50,7 +50,7 @@ class PreTrainedPipeline():
|
|
| 50 |
Return:
|
| 51 |
A :obj:`PIL.Image` with the raw image representation as PIL.
|
| 52 |
"""
|
| 53 |
-
noise = torch.randn(1, 100, 1, 1
|
| 54 |
with torch.no_grad():
|
| 55 |
output = self.model(noise)
|
| 56 |
|
|
|
|
| 50 |
Return:
|
| 51 |
A :obj:`PIL.Image` with the raw image representation as PIL.
|
| 52 |
"""
|
| 53 |
+
noise = torch.randn(1, 100, 1, 1)
|
| 54 |
with torch.no_grad():
|
| 55 |
output = self.model(noise)
|
| 56 |
|