Spaces:
Running
on
Zero
Running
on
Zero
Update app_v2v.py
Browse files- 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)
|