Spaces:
Running
on
Zero
Running
on
Zero
another fix
Browse files
model.py
CHANGED
|
@@ -829,7 +829,7 @@ class Model:
|
|
| 829 |
|
| 830 |
seq_1st = torch.tensor(seq_1st, dtype=torch.long,
|
| 831 |
device=self.device).unsqueeze(0)
|
| 832 |
-
if
|
| 833 |
output_list_1st = []
|
| 834 |
for tim in range(max(batch_size // mbz, 1)):
|
| 835 |
start_time = time.perf_counter()
|
|
|
|
| 829 |
|
| 830 |
seq_1st = torch.tensor(seq_1st, dtype=torch.long,
|
| 831 |
device=self.device).unsqueeze(0)
|
| 832 |
+
if image_prompt is None:
|
| 833 |
output_list_1st = []
|
| 834 |
for tim in range(max(batch_size // mbz, 1)):
|
| 835 |
start_time = time.perf_counter()
|