groggy84 commited on
Commit
ebb1911
1 Parent(s): c549bff
Files changed (3) hide show
  1. __pycache__/app.cpython-38.pyc +0 -0
  2. app.py +1 -1
  3. 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 = StableDiffusionControlNetPipeline.from_pretrained(
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