Spaces:
Runtime error
Runtime error
Commit
·
ea06b7b
1
Parent(s):
4cbb572
Update app.py
Browse files
app.py
CHANGED
@@ -11,13 +11,12 @@ def unclip_text_interpolation(
|
|
11 |
model_path,
|
12 |
start_prompt,
|
13 |
end_prompt,
|
14 |
-
steps
|
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
|
21 |
|
22 |
return images
|
23 |
|
|
|
11 |
model_path,
|
12 |
start_prompt,
|
13 |
end_prompt,
|
14 |
+
steps
|
|
|
15 |
):
|
16 |
|
17 |
pipe = DiffusionPipeline.from_pretrained(model_path, torch_dtype=torch.bfloat16, custom_pipeline='unclip_text_interpolation')
|
18 |
|
19 |
+
images = pipe(start_prompt, end_prompt, steps)
|
20 |
|
21 |
return images
|
22 |
|