prithivMLmods commited on
Commit
2afb41b
1 Parent(s): 7087e5f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -7,15 +7,15 @@ import torch
7
  from PIL import Image
8
 
9
  device = "cuda" if torch.cuda.is_available() else "cpu"
10
- model_repo_id = "stabilityai/stable-diffusion-3.5-large-turbo"
11
 
12
  torch_dtype = torch.bfloat16 if torch.cuda.is_available() else torch.float32
13
 
14
  pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
15
  pipe = pipe.to(device)
16
 
17
- pipe.load_lora_weights("prithivMLmods/SD3.5-Large-Turbo-HyperRealistic-LoRA", weight_name="SD3.5-4Step-Large-Turbo-HyperRealistic-LoRA.safetensors")
18
- trigger_word = "hyper realistic" # Specify trigger word for LoRA
19
  pipe.fuse_lora(lora_scale=1.0)
20
 
21
  MAX_SEED = np.iinfo(np.int32).max
 
7
  from PIL import Image
8
 
9
  device = "cuda" if torch.cuda.is_available() else "cpu"
10
+ model_repo_id = "stabilityai/stable-diffusion-3.5-large"
11
 
12
  torch_dtype = torch.bfloat16 if torch.cuda.is_available() else torch.float32
13
 
14
  pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
15
  pipe = pipe.to(device)
16
 
17
+ pipe.load_lora_weights("prithivMLmods/SD3.5-Large-Anime-LoRA", weight_name="SD3.5-Large-Anime-LoRA.safetensors")
18
+ trigger_word = "Anime 35" # Specify trigger word for LoRA
19
  pipe.fuse_lora(lora_scale=1.0)
20
 
21
  MAX_SEED = np.iinfo(np.int32).max