Hugging Face's logo

Spaces:
mrfakename
/
Kokoro-API-2
Sleeping

App Files Community
yaron123 commited on
Commit
ffe6ef9
·
1 Parent(s): e611e71

commit

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -124,7 +124,9 @@ function custom(){
124
 
125
  # torch pipes
126
 
127
- image_pipe = FluxPipeline.from_pretrained(base, torch_dtype=torch.float32).to(device)
 
 
128
 
129
  video_pipe = CogVideoXImageToVideoPipeline.from_pretrained(
130
  "THUDM/CogVideoX-5b-I2V",
 
124
 
125
  # torch pipes
126
 
127
+ image_pipe = FluxPipeline.from_pretrained(base, torch_dtype=torch.bfloat16).to(device)
128
+
129
+ image_pipe.enable_model_cpu_offload()
130
 
131
  video_pipe = CogVideoXImageToVideoPipeline.from_pretrained(
132
  "THUDM/CogVideoX-5b-I2V",