Jordan Legg commited on
Commit
c230ecc
1 Parent(s): 511af21

added different lora

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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
- # trigger_word = "Ultra realistic"
15
- # pipe.load_lora_weights("prithivMLmods/Canopus-LoRA-Flux-UltraRealism-2.0")
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 cinematic still 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)
 
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)