| import sys | |
| import re | |
| class StderrInterceptor: | |
| def __init__(self, original): | |
| self.original = original | |
| def write(self, s): | |
| self.original.write(s) | |
| match = re.search(r'(\d+)%\|', s) | |
| if match: | |
| self.original.write(f"\nINTERCEPTED: {match.group(1)}%\n") | |
| self.original.flush() | |
| def flush(self): | |
| self.original.flush() | |
| sys.stderr = StderrInterceptor(sys.stderr) | |
| from transformers import AutoConfig | |
| AutoConfig.from_pretrained("gpt2", force_download=True) | |