temporarily disable langchain stuff in webui (#1719)

its breaking the exe

apps/stable_diffusion/scripts/train_lora_word.py

@@ -136,10 +136,7 @@ class LoraDataset(Dataset):
 
         if self.center_crop:
             crop = min(img.shape[0], img.shape[1])
-            (
-                h,
-                w,
-            ) = (
+            (h, w,) = (
                 img.shape[0],
                 img.shape[1],
             )

apps/stable_diffusion/shark_studio_imports.py

@@ -50,6 +50,7 @@ datas += collect_data_files("sentencepiece")
 datas += collect_data_files("jsonschema")
 datas += collect_data_files("jsonschema_specifications")
 datas += collect_data_files("cpuinfo")
+datas += collect_data_files("langchain")
 datas += [
     ("src/utils/resources/prompts.json", "resources"),
     ("src/utils/resources/model_db.json", "resources"),

apps/stable_diffusion/src/utils/utils.py

@@ -144,10 +144,7 @@ def compile_through_fx(
     if "cuda" in args.device:
         shark_args.enable_tf32 = True
 
-    (
-        mlir_module,
-        func_name,
-    ) = import_with_fx(
+    (mlir_module, func_name,) = import_with_fx(
         model=model,
         inputs=inputs,
         is_f16=is_f16,

apps/stable_diffusion/web/index.py

@@ -115,8 +115,8 @@ if __name__ == "__main__":
         txt2img_sendto_inpaint,
         txt2img_sendto_outpaint,
         txt2img_sendto_upscaler,
-        h2ogpt_upload,
-        h2ogpt_web,
+        # h2ogpt_upload,
+        # h2ogpt_web,
         img2img_web,
         img2img_custom_model,
         img2img_hf_model_id,
@@ -248,10 +248,10 @@ if __name__ == "__main__":
                         upscaler_status,
                     ]
                 )
-            with gr.TabItem(label="DocuChat Upload", id=11):
-                h2ogpt_upload.render()
-            with gr.TabItem(label="DocuChat(Experimental)", id=12):
-                h2ogpt_web.render()
+            # with gr.TabItem(label="DocuChat Upload", id=11):
+            #    h2ogpt_upload.render()
+            # with gr.TabItem(label="DocuChat(Experimental)", id=12):
+            #    h2ogpt_web.render()
 
         # send to buttons
         register_button_click(

apps/stable_diffusion/web/ui/__init__.py

@@ -79,7 +79,6 @@ from apps.stable_diffusion.web.ui.stablelm_ui import (
     llm_chat_api,
 )
 from apps.stable_diffusion.web.ui.generate_config import model_config_web
-from apps.stable_diffusion.web.ui.h2ogpt import h2ogpt_upload, h2ogpt_web
 from apps.stable_diffusion.web.ui.minigpt4_ui import minigpt4_web
 from apps.stable_diffusion.web.ui.outputgallery_ui import (
     outputgallery_web,

requirements.txt

@@ -35,6 +35,7 @@ py-cpuinfo
 tiktoken # for codegen
 joblib # for langchain
 timm # for MiniGPT4
+langchain
 
 # Keep PyInstaller at the end. Sometimes Windows Defender flags it but most folks can continue even if it errors
 pefile