Hugging Face's logo

Spaces:
kadirnar
/
ComfyUI-Demo
Running

App Files Community
31
kadirnar commited on
Commit
b31f197
·
verified ·
1 Parent(s): edf4c40

Update stable_cascade.py

Browse files
Files changed (1) hide show
  1. stable_cascade.py +1 -0
stable_cascade.py CHANGED
@@ -3,6 +3,7 @@ from diffusers import StableCascadeDecoderPipeline, StableCascadePriorPipeline
3
  import gradio as gr
4
 
5
  torch._dynamo.config.suppress_errors = True
 
6
 
7
  prior = StableCascadePriorPipeline.from_pretrained("stabilityai/stable-cascade-prior", torch_dtype=torch.bfloat16)
8
  decoder = StableCascadeDecoderPipeline.from_pretrained("stabilityai/stable-cascade", torch_dtype=torch.bfloat16)
 
3
  import gradio as gr
4
 
5
  torch._dynamo.config.suppress_errors = True
6
+ os.environ['TOKENIZERS_PARALLELISM'] = 'false'
7
 
8
  prior = StableCascadePriorPipeline.from_pretrained("stabilityai/stable-cascade-prior", torch_dtype=torch.bfloat16)
9
  decoder = StableCascadeDecoderPipeline.from_pretrained("stabilityai/stable-cascade", torch_dtype=torch.bfloat16)