RageshAntony commited on
Commit
d112f60
·
verified ·
1 Parent(s): 6bcec5f
Files changed (1) hide show
  1. check_app.py +1 -3
check_app.py CHANGED
@@ -13,9 +13,6 @@ from diffusers import (
13
  )
14
  import gradio as gr
15
 
16
- print("PIPE ATTR")
17
- print(signature(AuraFlowPipeline))
18
- print(signature(LuminaText2ImgPipeline))
19
  cache_dir = '/workspace/hf_cache'
20
 
21
  MODEL_CONFIGS = {
@@ -105,6 +102,7 @@ def create_pipeline_logic(prompt_text, model_name):
105
  #cache_dir=config["cache_dir"],
106
  torch_dtype=torch.bfloat16
107
  ).to("cuda")
 
108
  image = generate_image_with_progress(
109
  pipe, prompt_text, num_steps=num_steps, guidance_scale=guidance_scale, seed=seed, progress=progress
110
  )
 
13
  )
14
  import gradio as gr
15
 
 
 
 
16
  cache_dir = '/workspace/hf_cache'
17
 
18
  MODEL_CONFIGS = {
 
102
  #cache_dir=config["cache_dir"],
103
  torch_dtype=torch.bfloat16
104
  ).to("cuda")
105
+ print(f"PIPE ATTR {signature(pipe)}")
106
  image = generate_image_with_progress(
107
  pipe, prompt_text, num_steps=num_steps, guidance_scale=guidance_scale, seed=seed, progress=progress
108
  )