update
Browse files
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
|
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)
|