Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -25,7 +25,7 @@ MAX_IMAGE_SIZE = 2048
|
|
| 25 |
# Ensure model and scheduler are initialized in GPU-enabled function
|
| 26 |
if torch.cuda.is_available():
|
| 27 |
transformer = FluxTransformer2DModel.from_single_file(
|
| 28 |
-
"https://huggingface.co/ekt1701/Test_case/blob/main/
|
| 29 |
torch_dtype=torch.bfloat16
|
| 30 |
)
|
| 31 |
pipe = FluxPipeline.from_pretrained(
|
|
@@ -67,7 +67,7 @@ css="""
|
|
| 67 |
with gr.Blocks(css=css) as demo:
|
| 68 |
|
| 69 |
with gr.Column(elem_id="col-container"):
|
| 70 |
-
gr.HTML("<h1><center>Flux Playground</center></h1><p><center>
|
| 71 |
|
| 72 |
with gr.Row():
|
| 73 |
|
|
@@ -110,7 +110,7 @@ with gr.Blocks(css=css) as demo:
|
|
| 110 |
minimum=1,
|
| 111 |
maximum=50,
|
| 112 |
step=1,
|
| 113 |
-
value=
|
| 114 |
)
|
| 115 |
|
| 116 |
guidance_scale = gr.Slider(
|
|
|
|
| 25 |
# Ensure model and scheduler are initialized in GPU-enabled function
|
| 26 |
if torch.cuda.is_available():
|
| 27 |
transformer = FluxTransformer2DModel.from_single_file(
|
| 28 |
+
"https://huggingface.co/ekt1701/Test_case/blob/main/artsyVibe_v2FP8.safetensors",
|
| 29 |
torch_dtype=torch.bfloat16
|
| 30 |
)
|
| 31 |
pipe = FluxPipeline.from_pretrained(
|
|
|
|
| 67 |
with gr.Blocks(css=css) as demo:
|
| 68 |
|
| 69 |
with gr.Column(elem_id="col-container"):
|
| 70 |
+
gr.HTML("<h1><center>Flux Playground</center></h1><p><center>Artsy Vibe v2 Model.</center></p>")
|
| 71 |
|
| 72 |
with gr.Row():
|
| 73 |
|
|
|
|
| 110 |
minimum=1,
|
| 111 |
maximum=50,
|
| 112 |
step=1,
|
| 113 |
+
value=24,
|
| 114 |
)
|
| 115 |
|
| 116 |
guidance_scale = gr.Slider(
|