Spaces:
Runtime error
Runtime error
Ritesh Khanna
commited on
Commit
·
cbc1cc3
1
Parent(s):
8cc6050
Add log to see why HF is hanging
Browse files
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)
|