AlexWortega

/

svggpt

like 0

An example of how to run would be great. I tried:
prompt = "three circles in a row colored red green blue"
tokenizer = AutoTokenizer.from_pretrained("AlexWortega/svggpt")
inputs = tokenizer(prompt, return_tensors="pt").input_ids
from transformers import AutoModelForCausalLM

model = AutoModelForCausalLM.from_pretrained('AlexWortega/svggpt')

outputs = model.generate(inputs, max_new_tokens=100, do_sample=True, top_k=50, top_p=0.95)
tokenizer.batch_decode(outputs, skip_special_tokens=True)

and got what looks like tokens for SVG:
'three circles in a row colored red green blue<|Y6|><|END|><|X115|><|Y0|><|BEZIER|><|START|><|X115|><|Y0|><|CONTROL1|><|X112|><|Y0|><|CONTROL2|><|X109|><|Y0|><|END|><|X107|><|Y2|><|BEZIER|><|START|><|X107|><|Y2|><|CONTROL1|><|X103|><|Y5|><|CONTROL2|><|X96|><|Y8|><|END|><|X90|><|Y13|><|BEZIER|><|START|><|X90|><|Y13|><|CONTROL1|><|X90|><|Y13|><|CONTROL2|><|X90|><|Y13|><|END|><|X90|><|Y13|><|LINE|><|START|><|X90|><|Y13|><|END|><|X83|><|Y20|><|BEZIER|><|START|><|X83|><|Y20|><|CONTROL1|><|X83|><|Y20|><|CONTROL2|><|X83|><|Y20|><|END|><|X83|><|Y20|><|LINE|><|START|><|X83|><|Y20|><|END|><|X92|><|Y31|><|BEZIER|><|START|><|X92|><|Y31|><|CONTROL1|><|X91|><|Y32|><|CONTROL2|><|X90|><|Y35|><|END|><|X89|><|Y36|><|LINE|><|START|><|X89|><|Y36|><|END|><|X86|><|Y39|><|LINE|><|START|><|X86|><|Y39|><|END|><|X87|><|Y39|><|LINE|><|START|><|X87|>'

So how do I get the SVG?