theostos commited on
Commit
a64ef34
1 Parent(s): a52d777

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 masked_llm.models.tokenizer_utils import generate_custom_mask, prepare_tokenizer
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