Ritesh Khanna commited on
Commit
cbc1cc3
1 Parent(s): 8cc6050

Add log to see why HF is hanging

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -69,6 +69,7 @@ def generate_text(prompt, extra=False, top_k=100, top_p=0.95, temperature=0.85,
69
  samples = []
70
  try:
71
  for i in range(1):
 
72
  outputs = model.generate(**inputs, max_length=256, do_sample=True, top_k=top_k, top_p=top_p, temperature=temperature, num_return_sequences=4, pad_token_id=tokenizer.eos_token_id)
73
  for output in outputs:
74
  sample = tokenizer.decode(output, skip_special_tokens=True)
 
69
  samples = []
70
  try:
71
  for i in range(1):
72
+ print(f"Generating sample {i+1}")
73
  outputs = model.generate(**inputs, max_length=256, do_sample=True, top_k=top_k, top_p=top_p, temperature=temperature, num_return_sequences=4, pad_token_id=tokenizer.eos_token_id)
74
  for output in outputs:
75
  sample = tokenizer.decode(output, skip_special_tokens=True)