spgoodman
commited on
Commit
·
f676c2e
1
Parent(s):
090d43a
Updated repo name
Browse files
README.md
CHANGED
|
@@ -40,8 +40,8 @@ from transformers import AutoProcessor, AutoModelForCausalLM
|
|
| 40 |
device = "cuda:0" if torch.cuda.is_available() else "cpu"
|
| 41 |
torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
|
| 42 |
|
| 43 |
-
model = AutoModelForCausalLM.from_pretrained("/Florence-2-base-PromptGen-v1.5", torch_dtype=torch_dtype, trust_remote_code=True).to(device)
|
| 44 |
-
processor = AutoProcessor.from_pretrained("/Florence-2-base-PromptGen-v1.5", trust_remote_code=True)
|
| 45 |
|
| 46 |
# Examples include CAPTION>, <DETAILED_CAPTION>, <MORE_DETAILED_CAPTION>,<GENERATE_TAGS>, <MIXED_CAPTION>, <0D>
|
| 47 |
prompt = "<CAPTION>"
|
|
|
|
| 40 |
device = "cuda:0" if torch.cuda.is_available() else "cpu"
|
| 41 |
torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
|
| 42 |
|
| 43 |
+
model = AutoModelForCausalLM.from_pretrained("createveai/Florence-2-base-PromptGen-v1.5", torch_dtype=torch_dtype, trust_remote_code=True).to(device)
|
| 44 |
+
processor = AutoProcessor.from_pretrained("createveai/Florence-2-base-PromptGen-v1.5", trust_remote_code=True)
|
| 45 |
|
| 46 |
# Examples include CAPTION>, <DETAILED_CAPTION>, <MORE_DETAILED_CAPTION>,<GENERATE_TAGS>, <MIXED_CAPTION>, <0D>
|
| 47 |
prompt = "<CAPTION>"
|