Spaces:
Sleeping
Sleeping
fix relative path
Browse files
app.py
CHANGED
|
@@ -5,7 +5,7 @@ import torch
|
|
| 5 |
|
| 6 |
from transformers import AutoTokenizer
|
| 7 |
from model.modeling_llamask import LlamaskForCausalLM
|
| 8 |
-
from
|
| 9 |
|
| 10 |
model_id = "meta-llama/Meta-Llama-3.1-8B-Instruct"
|
| 11 |
device = 'cpu'
|
|
|
|
| 5 |
|
| 6 |
from transformers import AutoTokenizer
|
| 7 |
from model.modeling_llamask import LlamaskForCausalLM
|
| 8 |
+
from model.tokenizer_utils import generate_custom_mask, prepare_tokenizer
|
| 9 |
|
| 10 |
model_id = "meta-llama/Meta-Llama-3.1-8B-Instruct"
|
| 11 |
device = 'cpu'
|
model/__pycache__/modeling_llamask.cpython-310.pyc
ADDED
|
Binary file (2.85 kB). View file
|
|
|