silveroxides commited on
Commit
b326fe8
1 Parent(s): 5188b2b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -9,6 +9,7 @@ import torch
9
 
10
  device = "cuda" if torch.cuda.is_available() else "cpu"
11
  text_encoder_repo = "silveroxides/CLIP_L_Fur"
 
12
  text_encoder_3_repo = "silveroxides/t5xxl_flan_enc"
13
  model_repo_id = "stabilityai/stable-diffusion-3.5-large-turbo"
14
 
@@ -17,8 +18,9 @@ if torch.cuda.is_available():
17
  else:
18
  torch_dtype = torch.float32
19
  text_encoder = CLIPTextModelWithProjection.from_pretrained(text_encoder_repo, torch_dtype=torch_dtype)
 
20
  text_encoder_3 = T5EncoderModel.from_pretrained(text_encoder_3_repo, torch_dtype=torch_dtype)
21
- pipe = DiffusionPipeline.from_pretrained(model_repo_id, text_encoder=text_encoder, text_encoder_3=text_encoder_3, torch_dtype=torch_dtype)
22
  pipe = pipe.to(device)
23
 
24
  MAX_SEED = np.iinfo(np.int32).max
 
9
 
10
  device = "cuda" if torch.cuda.is_available() else "cpu"
11
  text_encoder_repo = "silveroxides/CLIP_L_Fur"
12
+ text_encoder_2_repo = "silveroxides/SeaArtFurryCLIP_G"
13
  text_encoder_3_repo = "silveroxides/t5xxl_flan_enc"
14
  model_repo_id = "stabilityai/stable-diffusion-3.5-large-turbo"
15
 
 
18
  else:
19
  torch_dtype = torch.float32
20
  text_encoder = CLIPTextModelWithProjection.from_pretrained(text_encoder_repo, torch_dtype=torch_dtype)
21
+ text_encoder_2 = CLIPTextModelWithProjection.from_pretrained(text_encoder_2_repo, torch_dtype=torch_dtype)
22
  text_encoder_3 = T5EncoderModel.from_pretrained(text_encoder_3_repo, torch_dtype=torch_dtype)
23
+ pipe = DiffusionPipeline.from_pretrained(model_repo_id, text_encoder=text_encoder, text_encoder_2=text_encoder_2, text_encoder_3=text_encoder_3, torch_dtype=torch_dtype)
24
  pipe = pipe.to(device)
25
 
26
  MAX_SEED = np.iinfo(np.int32).max