Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,6 @@ import torch
|
|
3 |
import os
|
4 |
import gradio as gr
|
5 |
import sentencepiece
|
6 |
-
from tokenization_yi import YiTokenizer
|
7 |
|
8 |
|
9 |
os.environ['PYTORCH_CUDA_ALLOC_CONF'] = 'max_split_size_mb:120'
|
|
|
3 |
import os
|
4 |
import gradio as gr
|
5 |
import sentencepiece
|
|
|
6 |
|
7 |
|
8 |
os.environ['PYTORCH_CUDA_ALLOC_CONF'] = 'max_split_size_mb:120'
|