Spaces:
Running
on
Zero
Running
on
Zero
prithivMLmods
commited on
Commit
•
fcc63e9
1
Parent(s):
b3928d6
Update app.py
Browse files
app.py
CHANGED
@@ -37,11 +37,14 @@ if not torch.cuda.is_available():
|
|
37 |
base_model = "black-forest-labs/FLUX.1-dev"
|
38 |
pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
|
39 |
|
40 |
-
lora_repo = "prithivMLmods/
|
41 |
-
trigger_word = "
|
42 |
pipe.load_lora_weights(lora_repo)
|
43 |
|
44 |
-
|
|
|
|
|
|
|
45 |
|
46 |
style_list = [
|
47 |
{
|
|
|
37 |
base_model = "black-forest-labs/FLUX.1-dev"
|
38 |
pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
|
39 |
|
40 |
+
lora_repo = "prithivMLmods/Fashion-Hut-Modeling-LoRA"
|
41 |
+
trigger_word = "Modeling of" # Leave trigger_word blank if not used.
|
42 |
pipe.load_lora_weights(lora_repo)
|
43 |
|
44 |
+
#lora_repo = "prithivMLmods/Canopus-LoRA-Flux-UltraRealism-2.0"
|
45 |
+
#trigger_word = "Ultra realistic" # Leave trigger_word blank if not used.
|
46 |
+
#pipe.load_lora_weights(lora_repo)
|
47 |
+
#pipe.to("cuda")
|
48 |
|
49 |
style_list = [
|
50 |
{
|