ovi054 commited on
Commit
9757f67
·
verified ·
1 Parent(s): 467243d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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(