float32
Browse files
app.py
CHANGED
|
@@ -150,7 +150,7 @@ pipeline = TwoStagePipeline(
|
|
| 150 |
stage1_sampler_config,
|
| 151 |
stage2_sampler_config,
|
| 152 |
device=args.device,
|
| 153 |
-
dtype=torch.
|
| 154 |
)
|
| 155 |
|
| 156 |
with gr.Blocks() as demo:
|
|
|
|
| 150 |
stage1_sampler_config,
|
| 151 |
stage2_sampler_config,
|
| 152 |
device=args.device,
|
| 153 |
+
dtype=torch.float32
|
| 154 |
)
|
| 155 |
|
| 156 |
with gr.Blocks() as demo:
|