Hatman
/

Flux-Prompt-Enhance

Text Generation

text2text-generation

text-generation-inference

Model card Files Files and versions

HatmanStack commited on Jan 31

Commit

c8d5c42

·

1 Parent(s): 4b05d70

checkpoint

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -15,7 +15,7 @@ from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
 device = "cuda" if torch.cuda.is_available() else "cpu"
 # Model checkpoint
-model_checkpoint = "gokaygokay/Flux-Prompt-Enhance"
 # Tokenizer
 tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
@@ -58,7 +58,7 @@ class PromptEnhancer:
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
         # Model checkpoint
-        self.model_checkpoint = "gokaygokay/Flux-Prompt-Enhance"
         # Tokenizer and Model
         self.tokenizer = AutoTokenizer.from_pretrained(self.model_checkpoint)

 device = "cuda" if torch.cuda.is_available() else "cpu"
 # Model checkpoint
+model_checkpoint = "Hatman/Flux-Prompt-Enhance"
 # Tokenizer
 tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
         # Model checkpoint
+        self.model_checkpoint = "Hatman/Flux-Prompt-Enhance"
         # Tokenizer and Model
         self.tokenizer = AutoTokenizer.from_pretrained(self.model_checkpoint)