Spaces:

wencheng256
/

DiffusionRAWSpace

Sleeping

wencheng256 commited on May 8

Commit

3b6559c

1 Parent(s): 5407ff1

cpu

Files changed (2) hide show

__pycache__/pipeline_controlnet_sd_xl_raw.cpython-310.pyc CHANGED Viewed

Binary files a/__pycache__/pipeline_controlnet_sd_xl_raw.cpython-310.pyc and b/__pycache__/pipeline_controlnet_sd_xl_raw.cpython-310.pyc differ

app.py CHANGED Viewed

@@ -9,10 +9,10 @@ import traceback
 # ========== 1. Load Models ==========
 pipe = StableDiffusionXLControlNetRAWPipeline.from_pretrained(
     "wencheng256/DiffusionRAW",
-    torch_dtype=torch.float16
 )
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
-pipe.enable_model_cpu_offload()
 # ========== 2. Utility function: tensor -> PIL ==========
 def tensor_to_pil(img_tensor: torch.Tensor) -> Image.Image:

 # ========== 1. Load Models ==========
 pipe = StableDiffusionXLControlNetRAWPipeline.from_pretrained(
     "wencheng256/DiffusionRAW",
+    torch_dtype=torch.float32
 )
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
+# pipe.enable_model_cpu_offload()
 # ========== 2. Utility function: tensor -> PIL ==========
 def tensor_to_pil(img_tensor: torch.Tensor) -> Image.Image: