ywlee88 commited on
Commit
2c809d7
·
verified ·
1 Parent(s): a4dcd96

Upload folder using huggingface_hub

Browse files
safellava/model/language_model/safe_llava_llama.py CHANGED
@@ -5,15 +5,15 @@ import torch.nn as nn
5
  from transformers import AutoConfig, AutoModelForCausalLM
6
  from transformers.modeling_outputs import CausalLMOutputWithPast
7
 
8
- from llava.model.language_model.llava_llama import (
9
  LlavaConfig, LlavaLlamaModel, LlavaLlamaForCausalLM
10
  )
11
- from llava.constants import IMAGE_TOKEN_INDEX
12
 
13
  from dataclasses import dataclass
14
 
15
  import logging
16
- from llava.utils import setup_simple_logging
17
 
18
  setup_simple_logging()
19
 
 
5
  from transformers import AutoConfig, AutoModelForCausalLM
6
  from transformers.modeling_outputs import CausalLMOutputWithPast
7
 
8
+ from safellava.model.language_model.llava_llama import (
9
  LlavaConfig, LlavaLlamaModel, LlavaLlamaForCausalLM
10
  )
11
+ from safellava.constants import IMAGE_TOKEN_INDEX
12
 
13
  from dataclasses import dataclass
14
 
15
  import logging
16
+ from safellava.utils import setup_simple_logging
17
 
18
  setup_simple_logging()
19