ifmain
/

blip-image2promt-stable-diffusion-base

image-text-to-text

Model card Files Files and versions

ifmain commited on Jul 25, 2024

Commit

6f4446c

·

verified ·

1 Parent(s): ca26771

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -31,7 +31,7 @@ def prepare(text):
     text = re.sub(r'<[^>]*>', '', text)
     return text
-path_to_model = "blip-image2promt-stable-diffusion-v0.15"
 processor = BlipProcessor.from_pretrained(path_to_model)
 model = BlipForConditionalGeneration.from_pretrained(path_to_model, torch_dtype=torch.float16).to("cuda")

     text = re.sub(r'<[^>]*>', '', text)
     return text
+path_to_model = "ifmain/blip-image2promt-stable-diffusion"
 processor = BlipProcessor.from_pretrained(path_to_model)
 model = BlipForConditionalGeneration.from_pretrained(path_to_model, torch_dtype=torch.float16).to("cuda")