Spaces:
Runtime error
Runtime error
Commit
·
4cbb572
1
Parent(s):
9253136
Update app.py
Browse files
app.py
CHANGED
@@ -15,7 +15,7 @@ def unclip_text_interpolation(
|
|
15 |
num_inference_steps
|
16 |
):
|
17 |
|
18 |
-
pipe = DiffusionPipeline.from_pretrained(
|
19 |
|
20 |
images = pipe(start_prompt, end_prompt, steps, num_inference_steps=num_inference_steps)
|
21 |
|
|
|
15 |
num_inference_steps
|
16 |
):
|
17 |
|
18 |
+
pipe = DiffusionPipeline.from_pretrained(model_path, torch_dtype=torch.bfloat16, custom_pipeline='unclip_text_interpolation')
|
19 |
|
20 |
images = pipe(start_prompt, end_prompt, steps, num_inference_steps=num_inference_steps)
|
21 |
|