Use active model to generate response in example on README (#269) (#271)

Co-authored-by: rmilleti <[email protected]>
huggingface · Apr 3, 2023 · a2749d9 · a2749d9
1 parent ed87942
commit a2749d9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/README.md b/README.md
@@ -78,7 +78,7 @@ query_txt = "This morning I went to the "
 query_tensor = tokenizer.encode(query_txt, return_tensors="pt")
 
 # get model response
-response_tensor  = respond_to_batch(model_ref, query_tensor)
+response_tensor  = respond_to_batch(model, query_tensor)
 
 # create a ppo trainer
 ppo_trainer = PPOTrainer(ppo_config, model, model_ref, tokenizer)