traverse/utils/sort_with_llm.py
2026-02-07 02:53:05 +01:00

8 lines
174 B
Python

from transformers import pipeline
MODEL = "google/gemma-3-1b-it"
generator = pipeline("text-generation", model=MODEL, device_map="auto")
def sort_with_llm(tags):
pass