Spaces:
Running
on
A10G
Running
on
A10G
Anonymous
commited on
Commit
•
ecc18cb
1
Parent(s):
b135a58
fix app
Browse files
app.py
CHANGED
|
@@ -141,6 +141,10 @@ def infer(prompt):
|
|
| 141 |
grid = (grid + 1.0) / 2.0
|
| 142 |
grid = (grid * 255).to(torch.uint8).permute(0, 2, 3, 1)
|
| 143 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 144 |
torchvision.io.write_video(
|
| 145 |
video_path,
|
| 146 |
grid,
|
|
|
|
| 141 |
grid = (grid + 1.0) / 2.0
|
| 142 |
grid = (grid * 255).to(torch.uint8).permute(0, 2, 3, 1)
|
| 143 |
|
| 144 |
+
print(grid.shape)
|
| 145 |
+
print(torch.max(grid))
|
| 146 |
+
print(torch.min(grid))
|
| 147 |
+
|
| 148 |
torchvision.io.write_video(
|
| 149 |
video_path,
|
| 150 |
grid,
|