nllg
/

detikzify-ds-1.3b

Text Generation

Model card Files Files and versions

potamides commited on Dec 4, 2024

Commit

829f275

·

verified ·

1 Parent(s): 94488cb

Update README.md

Files changed (1) hide show

README.md +2 -3

README.md CHANGED Viewed

@@ -23,13 +23,12 @@ from operator import itemgetter
 from detikzify.model import load
 from detikzify.infer import DetikzifyPipeline
-import torch
 image = "https://w.wiki/A7Cc"
 pipeline = DetikzifyPipeline(*load(
-    base_model="nllg/detikzify-ds-1.3b",
     device_map="auto",
-    torch_dtype=torch.bfloat16,
 ))
 # generate a single TikZ program

 from detikzify.model import load
 from detikzify.infer import DetikzifyPipeline
 image = "https://w.wiki/A7Cc"
 pipeline = DetikzifyPipeline(*load(
+    model_name_or_path="nllg/detikzify-ds-1.3b",
     device_map="auto",
+    torch_dtype="bfloat16",
 ))
 # generate a single TikZ program