tifa-benchmark
/

llama2_tifa_question_generation

Text Generation

text-generation-inference

Model card Files Files and versions

yushihu commited on Aug 23, 2023

Commit

ccc014d

·

1 Parent(s): fea323b

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -32,7 +32,7 @@ import torch
 import transformers
 # prepare the LLaMA 2 model
-model_name = "/gscratch/tial/yushihu/tifa-all/llama2/results/llama2/final_question_generation_checkpoint"
 pipeline = transformers.pipeline(
     "text-generation",
     model=model_name,

 import transformers
 # prepare the LLaMA 2 model
+model_name = "tifa-benchmark/llama2_tifa_question_generation"
 pipeline = transformers.pipeline(
     "text-generation",
     model=model_name,