initial add

app.py

@@ -48,15 +48,14 @@ class GlobalText:
         # config dirs
         self.basedir                = os.getcwd()
         self.stable_diffusion_dir   = os.path.join(self.basedir, "models", "StableDiffusion")
-        self.personalized_model_dir = '/home/jin.liu/liujin/webui/stable-diffusion-webui/models/Stable-diffusion'
-        self.lora_model_dir         = '/home/jin.liu/liujin/webui/stable-diffusion-webui/models/Lora'
+        self.personalized_model_dir = './models/Stable-diffusion'
+        self.lora_model_dir         = './models/Lora'
         self.savedir                = os.path.join(self.basedir, "samples", datetime.now().strftime("Gradio-%Y-%m-%dT%H-%M-%S"))
         self.savedir_sample         = os.path.join(self.savedir, "sample")
 
         self.savedir_mask         = os.path.join(self.savedir, "mask")
 
-        self.stable_diffusion_list   = ["/home/jin.liu/liujin/cache/hub/models--runwayml--stable-diffusion-v1-5/snapshots/1d0c4ebf6ff58a5caecab40fa1406526bca4b5b9",
-                                        "runwayml/stable-diffusion-v1-5",
+        self.stable_diffusion_list   = ["runwayml/stable-diffusion-v1-5",
                                         "stabilityai/stable-diffusion-2-1"]
         self.personalized_model_list = []
         self.lora_model_list = []

requirements.txt

@@ -3,4 +3,5 @@ transformers
 opencv-python
 einops
 omegaconf
-pytorch_lightning
+pytorch_lightning
+torchvision