Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
b5329fc
1
Parent(s):
2bbf193
Update app.py
Browse files
app.py
CHANGED
|
@@ -44,7 +44,7 @@ MAX_SEED = np.iinfo(np.int32).max
|
|
| 44 |
DEVICE = "cuda" #if torch.cuda.is_available() else "cpu"
|
| 45 |
|
| 46 |
inpaint_pipe = FluxInpaintPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16).to(DEVICE)
|
| 47 |
-
|
| 48 |
|
| 49 |
|
| 50 |
@spaces.GPU()
|
|
|
|
| 44 |
DEVICE = "cuda" #if torch.cuda.is_available() else "cpu"
|
| 45 |
|
| 46 |
inpaint_pipe = FluxInpaintPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16).to(DEVICE)
|
| 47 |
+
inpaint_pipe.load_lora_weights('hugovntr/flux-schnell-realism', weight_name='schnell-realism_v1')
|
| 48 |
|
| 49 |
|
| 50 |
@spaces.GPU()
|