Hugging Face's logo

Spaces:
blanchon
/
JanusPro
Paused

App Files Community
1
blanchon commited on
Commit
6c5a909
·
1 Parent(s): bc8a8fa

Revert back to eager

Browse files
Files changed (1) hide show
  1. demo/app_januspro.py +2 -2
demo/app_januspro.py CHANGED
@@ -10,8 +10,8 @@ from transformers import AutoConfig, AutoModelForCausalLM
10
  model_path = "deepseek-ai/Janus-Pro-7B"
11
  config = AutoConfig.from_pretrained(model_path)
12
  language_config = config.language_config
13
- # language_config._attn_implementation = "eager"
14
- language_config._attn_implementation = "flash_attention_2"
15
 
16
  vl_gpt = AutoModelForCausalLM.from_pretrained(
17
  model_path, language_config=language_config, trust_remote_code=True
 
10
  model_path = "deepseek-ai/Janus-Pro-7B"
11
  config = AutoConfig.from_pretrained(model_path)
12
  language_config = config.language_config
13
+ language_config._attn_implementation = "eager"
14
+ # language_config._attn_implementation = "flash_attention_2"
15
 
16
  vl_gpt = AutoModelForCausalLM.from_pretrained(
17
  model_path, language_config=language_config, trust_remote_code=True