Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -85,7 +85,7 @@ def generate(prompt, negative_prompt, width=1024, height=1024, num_inference_ste
|
|
85 |
print("Activating base LoRA only.")
|
86 |
pipe.set_adapters([DEFAULT_LORA_NAME], adapter_weights=[1.0])
|
87 |
|
88 |
-
|
89 |
#pipe.to("cuda")
|
90 |
# apply_first_block_cache(pipe.transformer, FirstBlockCacheConfig(threshold=0.2))
|
91 |
apply_cache_on_pipe(
|
|
|
85 |
print("Activating base LoRA only.")
|
86 |
pipe.set_adapters([DEFAULT_LORA_NAME], adapter_weights=[1.0])
|
87 |
|
88 |
+
pipe.to(device)
|
89 |
#pipe.to("cuda")
|
90 |
# apply_first_block_cache(pipe.transformer, FirstBlockCacheConfig(threshold=0.2))
|
91 |
apply_cache_on_pipe(
|