ci/qwen-image
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

autocast

Browse Source
This commit is contained in:
sigma
2026-04-13 00:04:20 +00:00
committed by system
parent 2f633b7f46
commit cbc2ac12dd
1 changed files with 14 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
app.py
View File

@@ -531,6 +531,8 @@ def infer(
print(f"Rewritten Prompt: {prompt}") print(f"Rewritten Prompt: {prompt}")
# Enable Autocast for better results.
with torch.autocast(device_type="cuda", dtype=torch.bfloat16):
# Generate the image # Generate the image
image = pipe( image = pipe(
image=pil_images if len(pil_images) > 0 else None, image=pil_images if len(pil_images) > 0 else None,