Hugging Face's logo

nllg
/
detikzify-ds-1.3b

Text Generation
Transformers
Safetensors
detikzify
Inference Endpoints
Model card Files Community
potamides commited on
Commit
829f275
·
verified ·
1 Parent(s): 94488cb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -3
README.md CHANGED
@@ -23,13 +23,12 @@ from operator import itemgetter
23
 
24
  from detikzify.model import load
25
  from detikzify.infer import DetikzifyPipeline
26
- import torch
27
 
28
  image = "https://w.wiki/A7Cc"
29
  pipeline = DetikzifyPipeline(*load(
30
- base_model="nllg/detikzify-ds-1.3b",
31
  device_map="auto",
32
- torch_dtype=torch.bfloat16,
33
  ))
34
 
35
  # generate a single TikZ program
 
23
 
24
  from detikzify.model import load
25
  from detikzify.infer import DetikzifyPipeline
 
26
 
27
  image = "https://w.wiki/A7Cc"
28
  pipeline = DetikzifyPipeline(*load(
29
+ model_name_or_path="nllg/detikzify-ds-1.3b",
30
  device_map="auto",
31
+ torch_dtype="bfloat16",
32
  ))
33
 
34
  # generate a single TikZ program