Spaces:
Runtime error
Runtime error
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 |
)
|