ifmain
/

blip-image2promt-stable-diffusion-base

image-text-to-text

Inference Endpoints

Model card Files Files and versions Community

ifmain commited on Aug 4

Commit

462976e

•

1 Parent(s): 27818d1

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -36,7 +36,7 @@ def prepare(text):
     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")

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