Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -14,6 +14,10 @@ MAX_SEED = np.iinfo(np.int32).max
|
|
| 14 |
MAX_IMAGE_SIZE = 2048
|
| 15 |
|
| 16 |
pipe = FluxPipeline.from_pretrained("Himanshu806/FluxHyperReal", torch_dtype=torch.bfloat16).to("cuda")
|
|
|
|
|
|
|
|
|
|
|
|
|
| 17 |
|
| 18 |
@spaces.GPU(durations=300)
|
| 19 |
def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidance_scale=3.5, num_inference_steps=28, progress=gr.Progress(track_tqdm=True)):
|
|
|
|
| 14 |
MAX_IMAGE_SIZE = 2048
|
| 15 |
|
| 16 |
pipe = FluxPipeline.from_pretrained("Himanshu806/FluxHyperReal", torch_dtype=torch.bfloat16).to("cuda")
|
| 17 |
+
pipe.load_lora_weights("Himanshu806/testLora")
|
| 18 |
+
# pipe.enable_sequential_cpu_offload()
|
| 19 |
+
# pipe.enable_fp16()
|
| 20 |
+
pipe.enable_lora()
|
| 21 |
|
| 22 |
@spaces.GPU(durations=300)
|
| 23 |
def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidance_scale=3.5, num_inference_steps=28, progress=gr.Progress(track_tqdm=True)):
|