remove redundant statement in generate.py (#1139)

Gasoonjia · web-flow · commit 5a2e81079b0f · 2024-09-12T19:10:24.000-07:00
* update install_requirement.sh

* bring torchao back to all platform

* add essential comment

* enable torchtune on macos

* reformat

* remove reduntant statement
diff --git a/torchchat/generate.py b/torchchat/generate.py
@@ -739,7 +739,7 @@ def chat(
         )
         for i in range(num_samples):
             device_sync(device=self.builder_args.device)
-            if i >= 0 and generator_args.chat_mode:
+            if generator_args.chat_mode:
                 prompt = input("User: ")
                 if prompt == "/bye":
                     print("Exiting Chat.\n")
@@ -784,7 +784,6 @@ def chat(
                     )
                     break
 
-            if generator_args.chat_mode and i >= 0:
                 print("Model: ", end="")
 
                 buffer = []

Original file line number	Diff line number	Diff line change
`@@ -739,7 +739,7 @@ def chat(`
`739`	`739`	`)`
`740`	`740`	`for i in range(num_samples):`
`741`	`741`	`device_sync(device=self.builder_args.device)`
`742`		`- if i >= 0 and generator_args.chat_mode:`
	`742`	`+ if generator_args.chat_mode:`
`743`	`743`	`prompt = input("User: ")`
`744`	`744`	`if prompt == "/bye":`
`745`	`745`	`print("Exiting Chat.\n")`
`@@ -784,7 +784,6 @@ def chat(`
`784`	`784`	`)`
`785`	`785`	`break`
`786`	`786`
`787`		`- if generator_args.chat_mode and i >= 0:`
`788`	`787`	`print("Model: ", end="")`
`789`	`788`
`790`	`789`	`buffer = []`