fix zero

app.py

@@ -24,10 +24,9 @@ def inference(rgb, subject, prompt, strength, seed):
     
     if model is None:
         from model_lib.modules import MoMA_main_modal
-        # model = MoMA_main_modal(args).to(device, dtype=torch.float16)
+        model = MoMA_main_modal(args).to(device, dtype=torch.float16)
     
-    # generated_image = model.generate_images(rgb, subject, prompt, strength=strength, seed=seed)
-    generated_image = rgb
+    generated_image = model.generate_images(rgb, subject, prompt, strength=strength, seed=seed)
 
     return generated_image
 

env_test.sh

@@ -0,0 +1,10 @@
+pip uninstall  flash-attn
+
+git clone https://github.com/haotian-liu/LLaVA.git
+cd LLaVA
+
+pip install --upgrade pip  # enable PEP 660 support
+pip install -e .
+
+pip install -e ".[train]"
+pip install flash-attn --no-build-isolation --no-cache-dir