ekt1701 commited on
Commit
fff7903
·
verified ·
1 Parent(s): 7667adb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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/midjourneyReplica_flux1Dev.safetensors",
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>MidJourney Replica Model.</center></p>")
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=20,
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(