1inkusFace commited on
Commit
119381d
·
verified ·
1 Parent(s): b08aa0f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -185,7 +185,7 @@ def generate_30(
185
  ):
186
  seed = random.randint(0, MAX_SEED)
187
  generator = torch.Generator(device='cuda').manual_seed(seed)
188
- pipe.text_encoder=text_encoder.to(device=device, dtype=torch.bfloat16)
189
  pipe.text_encoder_2=text_encoder_2.to(device=device, dtype=torch.bfloat16)
190
  options = {
191
  "prompt": [prompt],
@@ -227,7 +227,7 @@ def generate_60(
227
  ):
228
  seed = random.randint(0, MAX_SEED)
229
  generator = torch.Generator(device='cuda').manual_seed(seed)
230
- pipe.text_encoder=text_encoder.to(device=device, dtype=torch.bfloat16)
231
  pipe.text_encoder_2=text_encoder_2.to(device=device, dtype=torch.bfloat16)
232
  options = {
233
  "prompt": [prompt],
@@ -269,7 +269,7 @@ def generate_90(
269
  ):
270
  seed = random.randint(0, MAX_SEED)
271
  generator = torch.Generator(device='cuda').manual_seed(seed)
272
- pipe.text_encoder=text_encoder.to(device=device, dtype=torch.bfloat16)
273
  pipe.text_encoder_2=text_encoder_2.to(device=device, dtype=torch.bfloat16)
274
  options = {
275
  "prompt": [prompt],
 
185
  ):
186
  seed = random.randint(0, MAX_SEED)
187
  generator = torch.Generator(device='cuda').manual_seed(seed)
188
+ #pipe.text_encoder=text_encoder.to(device=device, dtype=torch.bfloat16)
189
  pipe.text_encoder_2=text_encoder_2.to(device=device, dtype=torch.bfloat16)
190
  options = {
191
  "prompt": [prompt],
 
227
  ):
228
  seed = random.randint(0, MAX_SEED)
229
  generator = torch.Generator(device='cuda').manual_seed(seed)
230
+ #pipe.text_encoder=text_encoder.to(device=device, dtype=torch.bfloat16)
231
  pipe.text_encoder_2=text_encoder_2.to(device=device, dtype=torch.bfloat16)
232
  options = {
233
  "prompt": [prompt],
 
269
  ):
270
  seed = random.randint(0, MAX_SEED)
271
  generator = torch.Generator(device='cuda').manual_seed(seed)
272
+ #pipe.text_encoder=text_encoder.to(device=device, dtype=torch.bfloat16)
273
  pipe.text_encoder_2=text_encoder_2.to(device=device, dtype=torch.bfloat16)
274
  options = {
275
  "prompt": [prompt],