Update src/pipeline.py
Browse files- src/pipeline.py +5 -5
src/pipeline.py
CHANGED
|
@@ -32,11 +32,11 @@ def load_pipeline(pipeline=None) -> StableDiffusionXLPipeline:
|
|
| 32 |
pipeline = compile_pipe(pipeline)
|
| 33 |
load_pipe(pipeline, dir="/home/sandbox/.cache/huggingface/hub/models--RobertML--cached-pipe-02/snapshots/58d70deae87034cce351b780b48841f9746d4ad7")
|
| 34 |
|
| 35 |
-
instance = get_instance(device)
|
| 36 |
-
mul = torch.nn.Parameter(torch.tensor(1.2, requires_grad=False, device=device))
|
| 37 |
-
sub = torch.nn.Parameter(torch.tensor(0.75, requires_grad=False, device=device))
|
| 38 |
-
scaling_factor = torch.nn.Parameter(torch.tensor(pipeline.vae.config.scaling_factor, requires_grad=False, device=device))
|
| 39 |
-
hook_pipe(pipeline, instance, mul, sub, scaling_factor)
|
| 40 |
|
| 41 |
for _ in range(1):
|
| 42 |
deepcache_output = pipeline(prompt="telestereography, unstrengthen, preadministrator, copatroness, hyperpersonal, paramountness, paranoid, guaniferous", output_type="pil", num_inference_steps=20)
|
|
|
|
| 32 |
pipeline = compile_pipe(pipeline)
|
| 33 |
load_pipe(pipeline, dir="/home/sandbox/.cache/huggingface/hub/models--RobertML--cached-pipe-02/snapshots/58d70deae87034cce351b780b48841f9746d4ad7")
|
| 34 |
|
| 35 |
+
# instance = get_instance(device)
|
| 36 |
+
# mul = torch.nn.Parameter(torch.tensor(1.2, requires_grad=False, device=device))
|
| 37 |
+
# sub = torch.nn.Parameter(torch.tensor(0.75, requires_grad=False, device=device))
|
| 38 |
+
# scaling_factor = torch.nn.Parameter(torch.tensor(pipeline.vae.config.scaling_factor, requires_grad=False, device=device))
|
| 39 |
+
# hook_pipe(pipeline, instance, mul, sub, scaling_factor)
|
| 40 |
|
| 41 |
for _ in range(1):
|
| 42 |
deepcache_output = pipeline(prompt="telestereography, unstrengthen, preadministrator, copatroness, hyperpersonal, paramountness, paranoid, guaniferous", output_type="pil", num_inference_steps=20)
|