Spaces:

calihyper
/

choosa

Sleeping

groggy84 commited on May 4, 2023

Commit

a061910

•

1 Parent(s): bf5ab0d

no half

Files changed (2) hide show

__pycache__/app.cpython-38.pyc CHANGED Viewed

Binary files a/__pycache__/app.cpython-38.pyc and b/__pycache__/app.cpython-38.pyc differ

app.py CHANGED Viewed

@@ -29,11 +29,11 @@ def predict(canny, lt, ht, prompt, neg_prompt, ins, gs, seed):
     np_image = np.concatenate([np_image, np_image, np_image], axis=2)
     canny_image = Image.fromarray(np_image)
     controlnet_repo_id = "calihyper/kor-portrait-controlnet"
-    controlnet = ControlNetModel.from_pretrained(controlnet_repo_id, torch_dtype=torch.float16)
     '''
     repo_id = "calihyper/trad-kor-landscape-black"
     pipe = StableDiffusionPipeline.from_pretrained(
-        repo_id, torch_dtype=torch.float16
     )
     generator = torch.manual_seed(seed)

     np_image = np.concatenate([np_image, np_image, np_image], axis=2)
     canny_image = Image.fromarray(np_image)
     controlnet_repo_id = "calihyper/kor-portrait-controlnet"
+    controlnet = ControlNetModel.from_pretrained(controlnet_repo_id)
     '''
     repo_id = "calihyper/trad-kor-landscape-black"
     pipe = StableDiffusionPipeline.from_pretrained(
+        repo_id
     )
     generator = torch.manual_seed(seed)