Update app.py
Browse files
app.py
CHANGED
|
@@ -7,6 +7,7 @@ from transformers import GPT2LMHeadModel, GPT2Tokenizer, AutoTokenizer, AutoMode
|
|
| 7 |
from huggingface_hub import HfFolder
|
| 8 |
from io import StringIO
|
| 9 |
from tqdm import tqdm
|
|
|
|
| 10 |
from accelerate import init_empty_weight
|
| 11 |
|
| 12 |
# Access the Hugging Face API token from environment variables
|
|
|
|
| 7 |
from huggingface_hub import HfFolder
|
| 8 |
from io import StringIO
|
| 9 |
from tqdm import tqdm
|
| 10 |
+
import accelerate
|
| 11 |
from accelerate import init_empty_weight
|
| 12 |
|
| 13 |
# Access the Hugging Face API token from environment variables
|