Spaces:
Running
on
Zero
Running
on
Zero
Jordan Legg
commited on
Commit
·
c230ecc
1
Parent(s):
511af21
added different lora
Browse files
app.py
CHANGED
|
@@ -11,8 +11,8 @@ device: str = "cuda" if torch.cuda.is_available() else "cpu"
|
|
| 11 |
|
| 12 |
|
| 13 |
pipe = DiffusionPipeline.from_pretrained("shuttleai/shuttle-3-diffusion", torch_dtype=dtype).to(device)
|
| 14 |
-
|
| 15 |
-
|
| 16 |
# Enable VAE tiling
|
| 17 |
pipe.vae.enable_tiling()
|
| 18 |
|
|
@@ -37,7 +37,7 @@ MIN_WIDTH = 512
|
|
| 37 |
MAX_WIDTH = 3072
|
| 38 |
STANDARD_WIDTH = 2048
|
| 39 |
STEP_WIDTH = 8
|
| 40 |
-
STYLE_PROMPT = "hyperrealistic widescreen anamorphic
|
| 41 |
|
| 42 |
def calculate_height(width: int, aspect_ratio: float) -> int:
|
| 43 |
height = int(width / aspect_ratio)
|
|
|
|
| 11 |
|
| 12 |
|
| 13 |
pipe = DiffusionPipeline.from_pretrained("shuttleai/shuttle-3-diffusion", torch_dtype=dtype).to(device)
|
| 14 |
+
trigger_word = "CNSTLL"
|
| 15 |
+
pipe.load_lora_weights("adirik/flux-cinestill")
|
| 16 |
# Enable VAE tiling
|
| 17 |
pipe.vae.enable_tiling()
|
| 18 |
|
|
|
|
| 37 |
MAX_WIDTH = 3072
|
| 38 |
STANDARD_WIDTH = 2048
|
| 39 |
STEP_WIDTH = 8
|
| 40 |
+
STYLE_PROMPT = "analog film, high resolution, cinestill 800t, hyperrealistic, widescreen, anamorphic, vignette, bokeh, film grain, dramatic lighting, epic composition, moody, detailed, super wide shot, atmospheric, backlit, soft light, "
|
| 41 |
|
| 42 |
def calculate_height(width: int, aspect_ratio: float) -> int:
|
| 43 |
height = int(width / aspect_ratio)
|