Spaces:
Running
on
Zero
Running
on
Zero
prithivMLmods
commited on
Commit
•
0ce8ff5
1
Parent(s):
9f675c8
Update app.py
Browse files
app.py
CHANGED
@@ -36,15 +36,10 @@ if not torch.cuda.is_available():
|
|
36 |
|
37 |
base_model = "black-forest-labs/FLUX.1-dev"
|
38 |
pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
|
39 |
-
|
40 |
-
|
41 |
-
trigger_word = "Modeling of" # Leave trigger_word blank if not used.
|
42 |
pipe.load_lora_weights(lora_repo)
|
43 |
pipe.to("cuda")
|
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 |
{
|
|
|
36 |
|
37 |
base_model = "black-forest-labs/FLUX.1-dev"
|
38 |
pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
|
39 |
+
lora_repo = "prithivMLmods/Canopus-LoRA-Flux-UltraRealism-2.0"
|
40 |
+
trigger_word = "Ultra realistic" # Leave trigger_word blank if not used.
|
|
|
41 |
pipe.load_lora_weights(lora_repo)
|
42 |
pipe.to("cuda")
|
|
|
|
|
|
|
|
|
43 |
|
44 |
style_list = [
|
45 |
{
|