akhaliq HF staff commited on
Commit
23fd98a
1 Parent(s): 888af9f

another fix

Browse files
Files changed (1) hide show
  1. model.py +1 -1
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 self.image_prompt is None:
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()