Spaces:
Sleeping
Sleeping
Upload better_transformer.py
Browse files- better_transformer.py +2 -0
better_transformer.py
CHANGED
@@ -359,6 +359,8 @@ def generate(model, tokenizer, device, method=None, k=None,
|
|
359 |
if deterministic is not None:
|
360 |
set_seed(deterministic)
|
361 |
|
|
|
|
|
362 |
if cond != "":
|
363 |
|
364 |
cond_tokens = tokenizer(cond).input_ids
|
|
|
359 |
if deterministic is not None:
|
360 |
set_seed(deterministic)
|
361 |
|
362 |
+
st.markdown(f"Deterministic: {deterministic}")
|
363 |
+
|
364 |
if cond != "":
|
365 |
|
366 |
cond_tokens = tokenizer(cond).input_ids
|