LPX55 commited on
Commit
cf746b6
·
verified ·
1 Parent(s): b9aaf3f

Update app_v2v.py

Browse files
Files changed (1) hide show
  1. app_v2v.py +1 -1
app_v2v.py CHANGED
@@ -84,7 +84,7 @@ if not high_vram:
84
  transformer.high_quality_fp32_output_for_inference = True
85
  print('transformer.high_quality_fp32_output_for_inference = True')
86
 
87
- transformer.to(dtype=torch.bfloat16)
88
  vae.to(dtype=torch.float16)
89
  image_encoder.to(dtype=torch.float16)
90
  text_encoder.to(dtype=torch.float16)
 
84
  transformer.high_quality_fp32_output_for_inference = True
85
  print('transformer.high_quality_fp32_output_for_inference = True')
86
 
87
+ # transformer.to(dtype=torch.bfloat16)
88
  vae.to(dtype=torch.float16)
89
  image_encoder.to(dtype=torch.float16)
90
  text_encoder.to(dtype=torch.float16)