Damian Stewart commited on
Commit
a07c108
·
1 Parent(s): 2c1839c
Files changed (1) hide show
  1. train.py +1 -1
train.py CHANGED
@@ -35,7 +35,7 @@ def validate(diffuser: StableDiffuser, finetuner: FineTunedModel,
35
  nsteps=50
36
  num_validation_prompts = validation_embeddings.shape[0] // 2
37
 
38
- for i in tqdm(range(num_validation_prompts))
39
  if training_should_cancel:
40
  print("cancel requested, bailing")
41
  return
 
35
  nsteps=50
36
  num_validation_prompts = validation_embeddings.shape[0] // 2
37
 
38
+ for i in tqdm(range(num_validation_prompts)):
39
  if training_should_cancel:
40
  print("cancel requested, bailing")
41
  return