Spaces:
Running
on
Zero
Running
on
Zero
pipe attr
Browse files- 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 |
)
|