Hugging Face's logo

Spaces:
Hatman
/
deepseek-janus
Runtime error

App Files Community
HatmanStack commited on
Commit
cc7322e
·
1 Parent(s): 9b6e50e

added flux local

Browse files
Files changed (2) hide show
  1. Flux-Prompt-Enhance +1 -0
  2. app.py +1 -1
Flux-Prompt-Enhance ADDED
@@ -0,0 +1 @@
 
 
1
+ Subproject commit c8d5c42eaa68c13c1fc243825fdc4a99285fad60
app.py CHANGED
@@ -10,7 +10,7 @@ import os
10
 
11
  token = os.environ.get('HF_TOKEN')
12
  cuda_device = 'cuda' if torch.cuda.is_available() else 'cpu'
13
- model_checkpoint = "Hatman/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
 
 
10
 
11
  token = os.environ.get('HF_TOKEN')
12
  cuda_device = 'cuda' if torch.cuda.is_available() else 'cpu'
13
+ model_checkpoint = "./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