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