Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -27,8 +27,11 @@ MAX_SEED = np.iinfo(np.int32).max
|
|
| 27 |
base_model = "black-forest-labs/FLUX.1-dev"
|
| 28 |
pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
|
| 29 |
|
| 30 |
-
lora_repo = "strangerzonehf/Flux-Super-Realism-LoRA"
|
| 31 |
-
trigger_word = "Super Realism" # Leave blank if not used
|
|
|
|
|
|
|
|
|
|
| 32 |
|
| 33 |
pipe.load_lora_weights(lora_repo)
|
| 34 |
pipe.to("cuda")
|
|
|
|
| 27 |
base_model = "black-forest-labs/FLUX.1-dev"
|
| 28 |
pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
|
| 29 |
|
| 30 |
+
#lora_repo = "strangerzonehf/Flux-Super-Realism-LoRA"
|
| 31 |
+
#trigger_word = "Super Realism" # Leave blank if not used
|
| 32 |
+
|
| 33 |
+
lora_repo = "strangerzonehf/Flux-Midjourney-Studio-LoRA"
|
| 34 |
+
trigger_word = "mj x1" # Leave blank if not used
|
| 35 |
|
| 36 |
pipe.load_lora_weights(lora_repo)
|
| 37 |
pipe.to("cuda")
|