Spaces:
Runtime error
Runtime error
initial add
Browse files- app.py +3 -4
- requirements.txt +2 -1
app.py
CHANGED
@@ -48,15 +48,14 @@ class GlobalText:
|
|
48 |
# config dirs
|
49 |
self.basedir = os.getcwd()
|
50 |
self.stable_diffusion_dir = os.path.join(self.basedir, "models", "StableDiffusion")
|
51 |
-
self.personalized_model_dir = '
|
52 |
-
self.lora_model_dir = '
|
53 |
self.savedir = os.path.join(self.basedir, "samples", datetime.now().strftime("Gradio-%Y-%m-%dT%H-%M-%S"))
|
54 |
self.savedir_sample = os.path.join(self.savedir, "sample")
|
55 |
|
56 |
self.savedir_mask = os.path.join(self.savedir, "mask")
|
57 |
|
58 |
-
self.stable_diffusion_list = ["/
|
59 |
-
"runwayml/stable-diffusion-v1-5",
|
60 |
"stabilityai/stable-diffusion-2-1"]
|
61 |
self.personalized_model_list = []
|
62 |
self.lora_model_list = []
|
|
|
48 |
# config dirs
|
49 |
self.basedir = os.getcwd()
|
50 |
self.stable_diffusion_dir = os.path.join(self.basedir, "models", "StableDiffusion")
|
51 |
+
self.personalized_model_dir = './models/Stable-diffusion'
|
52 |
+
self.lora_model_dir = './models/Lora'
|
53 |
self.savedir = os.path.join(self.basedir, "samples", datetime.now().strftime("Gradio-%Y-%m-%dT%H-%M-%S"))
|
54 |
self.savedir_sample = os.path.join(self.savedir, "sample")
|
55 |
|
56 |
self.savedir_mask = os.path.join(self.savedir, "mask")
|
57 |
|
58 |
+
self.stable_diffusion_list = ["runwayml/stable-diffusion-v1-5",
|
|
|
59 |
"stabilityai/stable-diffusion-2-1"]
|
60 |
self.personalized_model_list = []
|
61 |
self.lora_model_list = []
|
requirements.txt
CHANGED
@@ -3,4 +3,5 @@ transformers
|
|
3 |
opencv-python
|
4 |
einops
|
5 |
omegaconf
|
6 |
-
pytorch_lightning
|
|
|
|
3 |
opencv-python
|
4 |
einops
|
5 |
omegaconf
|
6 |
+
pytorch_lightning
|
7 |
+
torchvision
|