WYBar commited on
Commit
907ad49
·
1 Parent(s): 6ae4c78
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -450,7 +450,7 @@ def test_one_sample(validation_box, validation_prompt, true_gs, inference_steps,
450
 
451
  return output_gradio
452
 
453
- def svg_test_one_sample(validation_prompt, validation_box_str, seed, true_gs, inference_steps, pipeline, transp_vae):
454
  print(f"svg_test_one_sample {model.device} {model.lm.device} {pipeline.device}")
455
  # generator = torch.Generator().manual_seed(seed)
456
  generator = torch.Generator(device=torch.device("cuda", index=0)).manual_seed(seed)
 
450
 
451
  return output_gradio
452
 
453
+ def gradio_test_one_sample(validation_prompt, validation_box_str, seed, true_gs, inference_steps, pipeline, transp_vae):
454
  print(f"svg_test_one_sample {model.device} {model.lm.device} {pipeline.device}")
455
  # generator = torch.Generator().manual_seed(seed)
456
  generator = torch.Generator(device=torch.device("cuda", index=0)).manual_seed(seed)