Kc-12 commited on
Commit
6a91a8f
1 Parent(s): 5047736

Upload better_transformer.py

Browse files
Files changed (1) hide show
  1. 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