Spaces:
Running
on
Zero
Running
on
Zero
Update controlnet/callable_functions.py
Browse files
controlnet/callable_functions.py
CHANGED
@@ -34,7 +34,7 @@ def process_single_image(model,image_path, prompt, num_inference_steps, stylecod
|
|
34 |
stylecodes_model=stylecodes_model,
|
35 |
torch_dtype=torch.float16,
|
36 |
device="cuda",
|
37 |
-
scheduler=noise_scheduler,
|
38 |
feature_extractor=None,
|
39 |
safety_checker=None,
|
40 |
)
|
|
|
34 |
stylecodes_model=stylecodes_model,
|
35 |
torch_dtype=torch.float16,
|
36 |
device="cuda",
|
37 |
+
#scheduler=noise_scheduler,
|
38 |
feature_extractor=None,
|
39 |
safety_checker=None,
|
40 |
)
|