Update modules/pmbl.py

modules/pmbl.py

@@ -103,15 +103,15 @@ class PMBL:
 
     def generate_response_task(self, system_prompt, prompt, n_ctx):
         def generate_response_task(self, system_prompt, prompt, n_ctx):
-        llm = Llama(
-            model_path=self.model_path,
-            n_ctx=n_ctx,
-            n_threads=8,
-            n_gpu_layers=-1,
-            use_mlock=True,
-            use_mmap=True,
-            use_flash_attn=True
-        )
+            llm = Llama(
+                model_path=self.model_path,
+                n_ctx=n_ctx,
+                n_threads=8,
+                n_gpu_layers=-1,
+                use_mlock=True,
+                use_mmap=True,
+                use_flash_attn=True
+            )
 
         response = llm.generate(
             system_prompt,
@@ -156,12 +156,12 @@ class PMBL:
         conn.close()
 
     def generate_topic(self, prompt, response):
-        llm = Llama(
-            model_path=self.model_path,
-            n_ctx=n_ctx,
-            n_threads=2,
-            n_gpu_layers=2,
-        )
+            llm = Llama(
+                model_path=self.model_path,
+                n_ctx=n_ctx,
+                n_threads=2,
+                n_gpu_layers=2,
+            )
 
         system_prompt = f"Based on the following interaction between a user and an AI assistant, generate a concise topic for the conversation in 2-4 words:\n\nUser: {prompt}\nAssistant: {response}\n\nTopic:"