From 2ee35fb8fe3e9528875da4aac94659ec5b1381e0 Mon Sep 17 00:00:00 2001
From: Jack-Khuu <jack.khuu.7@gmail.com>
Date: Sat, 3 Aug 2024 14:48:55 -0400
Subject: [PATCH] Avoid blindly accessing prompt and num_samples

---
 generate.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/generate.py b/generate.py
index 69d817723..eff086afd 100644
--- a/generate.py
+++ b/generate.py
@@ -110,11 +110,11 @@ def from_args(cls, args):
         )
 
         return cls(
-            prompt=args.prompt,
+            prompt=getattr(args, "prompt", ""),
             encoded_prompt=None,
             chat_mode=args.chat,
             gui_mode=args.gui,
-            num_samples=args.num_samples,
+            num_samples=getattr(args, "num_samples", 1),
             max_new_tokens=args.max_new_tokens,
             top_k=args.top_k,
             temperature=args.temperature,