Hugging Face's logo

Spaces:
cjgall
/
deepseek-janus-pro-7b
Running on Zero

App Files Community
cjgall commited on
Commit
b7cfdd4
·
verified ·
1 Parent(s): 8238723

update

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ model_checkpoint = "gokaygokay/Flux-Prompt-Enhance"
14
  enhancer_tokenizer = AutoTokenizer.from_pretrained(model_checkpoint, token=token)
15
  enhancer_model = AutoModelForSeq2SeqLM.from_pretrained(model_checkpoint).to(cuda_device)
16
 
17
- model_path = "deepseek/Janus-Pro-7B"
18
  config = AutoConfig.from_pretrained(model_path)
19
  language_config = config.language_config
20
  language_config._attn_implementation = 'eager'
 
14
  enhancer_tokenizer = AutoTokenizer.from_pretrained(model_checkpoint, token=token)
15
  enhancer_model = AutoModelForSeq2SeqLM.from_pretrained(model_checkpoint).to(cuda_device)
16
 
17
+ model_path = "deepseek-ai/Janus-Pro-7B"
18
  config = AutoConfig.from_pretrained(model_path)
19
  language_config = config.language_config
20
  language_config._attn_implementation = 'eager'