theostos commited on
Commit
a52d777
·
1 Parent(s): 44d58cf

fix relative path

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from huggingface_hub import InferenceClient
4
  import torch
5
 
6
  from transformers import AutoTokenizer
7
- from models.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"
 
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"