test edit
Browse files- __pycache__/app.cpython-38.pyc +0 -0
- app.py +1 -1
- requirements.txt +1 -0
__pycache__/app.cpython-38.pyc
CHANGED
Binary files a/__pycache__/app.cpython-38.pyc and b/__pycache__/app.cpython-38.pyc differ
|
|
app.py
CHANGED
@@ -32,7 +32,7 @@ def predict(canny, lt, ht, prompt, neg_prompt, ins, gs, seed):
|
|
32 |
controlnet = ControlNetModel.from_pretrained(controlnet_repo_id, torch_dtype=torch.float16)
|
33 |
'''
|
34 |
repo_id = "calihyper/trad-kor-landscape-black"
|
35 |
-
pipe =
|
36 |
repo_id, torch_dtype=torch.float16
|
37 |
)
|
38 |
generator = torch.manual_seed(seed)
|
|
|
32 |
controlnet = ControlNetModel.from_pretrained(controlnet_repo_id, torch_dtype=torch.float16)
|
33 |
'''
|
34 |
repo_id = "calihyper/trad-kor-landscape-black"
|
35 |
+
pipe = StableDiffusionPipeline.from_pretrained(
|
36 |
repo_id, torch_dtype=torch.float16
|
37 |
)
|
38 |
generator = torch.manual_seed(seed)
|
requirements.txt
CHANGED
@@ -5,6 +5,7 @@ gradio
|
|
5 |
Pillow
|
6 |
numpy
|
7 |
transformers
|
|
|
8 |
ftfy
|
9 |
altair
|
10 |
opencv-python
|
|
|
5 |
Pillow
|
6 |
numpy
|
7 |
transformers
|
8 |
+
torchvision
|
9 |
ftfy
|
10 |
altair
|
11 |
opencv-python
|