Spaces:
Running
Running
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()
|