Spaces:
Runtime error
Runtime error
Update demo_util.py
Browse files- demo_util.py +0 -8
demo_util.py
CHANGED
|
@@ -57,14 +57,6 @@ def sample_fn(generator,
|
|
| 57 |
randomize_temperature=2.0,
|
| 58 |
num_sample_steps=8,
|
| 59 |
device="cuda"):
|
| 60 |
-
print(f"generator:{generator}")
|
| 61 |
-
print(f"tokenizer:{tokenizer}")
|
| 62 |
-
print(f"labels:{labels}")
|
| 63 |
-
print(f"guidance_scale:{guidance_scale}")
|
| 64 |
-
print(f"randomize_temperature:{randomize_temperature}")
|
| 65 |
-
print(f"num_sample_steps:{num_sample_steps}")
|
| 66 |
-
print(f"device:{device}")
|
| 67 |
-
|
| 68 |
generator.eval()
|
| 69 |
tokenizer.eval()
|
| 70 |
if labels is None:
|
|
|
|
| 57 |
randomize_temperature=2.0,
|
| 58 |
num_sample_steps=8,
|
| 59 |
device="cuda"):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 60 |
generator.eval()
|
| 61 |
tokenizer.eval()
|
| 62 |
if labels is None:
|