Spaces:

Kjgarza
/

metadata_transformer

Paused

Kjgarza commited on Oct 17, 2023

Commit

a49498e

1 Parent(s): 231054f

pipeline

Files changed (1) hide show

metadata_transformer.py CHANGED Viewed

@@ -11,7 +11,7 @@ login(token=TOKEN)
 model = "meta-llama/Llama-2-7b-chat-hf"
 tokenizer = AutoTokenizer.from_pretrained(model)
-pipeline = pipeline(
     "text-generation",
     model=model,
     torch_dtype=torch.float16,

 model = "meta-llama/Llama-2-7b-chat-hf"
 tokenizer = AutoTokenizer.from_pretrained(model)
+pipeline = transformers.pipeline(
     "text-generation",
     model=model,
     torch_dtype=torch.float16,