Spaces:
Sleeping
Sleeping
fix relative path
Browse files
app.py
CHANGED
@@ -4,7 +4,7 @@ from huggingface_hub import InferenceClient
|
|
4 |
import torch
|
5 |
|
6 |
from transformers import AutoTokenizer
|
7 |
-
from
|
8 |
from masked_llm.models.tokenizer_utils import generate_custom_mask, prepare_tokenizer
|
9 |
|
10 |
model_id = "meta-llama/Meta-Llama-3.1-8B-Instruct"
|
|
|
4 |
import torch
|
5 |
|
6 |
from transformers import AutoTokenizer
|
7 |
+
from model.modeling_llamask import LlamaskForCausalLM
|
8 |
from masked_llm.models.tokenizer_utils import generate_custom_mask, prepare_tokenizer
|
9 |
|
10 |
model_id = "meta-llama/Meta-Llama-3.1-8B-Instruct"
|