Hugging Face's logo

nllg
/
detikzify-tl-1.1b

Text Generation
Transformers
Safetensors
detikzify
Inference Endpoints
Model card Files Community
potamides commited on
Commit
066cfd5
·
verified ·
1 Parent(s): 3b49f13

Update README.md

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