Update unidiffuser/sample_v1.py
Browse filesFix sample function for gradio app
- unidiffuser/sample_v1.py +2 -1
unidiffuser/sample_v1.py
CHANGED
@@ -433,5 +433,6 @@ def sample(mode, prompt, image, sample_steps=50, scale=7.0, seed=None):
|
|
433 |
if seed is not None:
|
434 |
config.seed = seed
|
435 |
|
436 |
-
evaluate(config)
|
|
|
437 |
|
|
|
433 |
if seed is not None:
|
434 |
config.seed = seed
|
435 |
|
436 |
+
sample_images, sample_text = evaluate(config)
|
437 |
+
return sample_images, sample_text
|
438 |
|