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
|
|