nllg
/

potamides commited on
Commit
1710da2
·
verified ·
1 Parent(s): ae3b65c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -23,9 +23,9 @@ import torch
23
 
24
  image = "https://w.wiki/A7Cc"
25
  pipeline = DetikzifyPipeline(*load(
26
- base_model="nllg/detikzify-cl-7b",
27
  device_map="auto",
28
- torch_dtype=torch.bfloat16,
29
  ))
30
 
31
  # generate a single TikZ program
 
23
 
24
  image = "https://w.wiki/A7Cc"
25
  pipeline = DetikzifyPipeline(*load(
26
+ model_name_or_path="nllg/detikzify-cl-7b",
27
  device_map="auto",
28
+ torch_dtype="bfloat16",
29
  ))
30
 
31
  # generate a single TikZ program