spgoodman commited on
Commit
f676c2e
·
1 Parent(s): 090d43a

Updated repo name

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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>"