Axel-Student commited on
Commit
5a644ca
·
1 Parent(s): 7857c96

remove useless pipe

Browse files
Files changed (1) hide show
  1. app.py +0 -3
app.py CHANGED
@@ -4,9 +4,6 @@ from diffusers import FluxPipeline # type: ignore
4
  import gradio as gr # type: ignore
5
  from huggingface_hub import login, InferenceClient
6
 
7
- pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
8
- pipe.enable_model_cpu_offload() #save some VRAM by offloading the model to CPU. Remove this if you have enough GPU power
9
-
10
  token = os.getenv("HF_TOKEN")
11
  login(token=token)
12
 
 
4
  import gradio as gr # type: ignore
5
  from huggingface_hub import login, InferenceClient
6
 
 
 
 
7
  token = os.getenv("HF_TOKEN")
8
  login(token=token)
9