commit

app.py

@@ -502,9 +502,8 @@ function custom(){
 # torch pipes
 
 image_pipe = FluxPipeline.from_pretrained(base, torch_dtype=torch.bfloat16).to(device)
-#image_pipe.enable_model_cpu_offload()
 
-pegasus_tokenizer = PegasusTokenizerFast.from_pretrained("google/pegasus-xsum"),
+pegasus_tokenizer = PegasusTokenizerFast.from_pretrained("google/pegasus-xsum")
 pegasus_model = PegasusForConditionalGeneration.from_pretrained("google/pegasus-xsum")
 
 # functionality