Hugging Face's logo

Spaces:
NagaSaiAbhinay
/
unclip_text_interpolation_demo
Runtime error

App Files Community
2
NagaSaiAbhinay commited on
Commit
929d65e
·
1 Parent(s): ea06b7b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ def unclip_text_interpolation(
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
 
 
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, enable_sequential_cpu_offload=False)
20
 
21
  return images
22