Spaces:

NJU
/

RAG-Diffusion

Running on Zero

znchen commited on Nov 19, 2024

Commit

64605b3

1 Parent(s): c0d06d1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ from RAG_pipeline_flux import RAG_FluxPipeline
 MAX_SEED = 999999
-pipe = RAG_FluxPipeline.from_pretrained("/nasdata/znchen/nju_reseach/FLUX.1-dev/", torch_dtype=torch.bfloat16)
 pipe = pipe.to("cuda")
 global run_nums

 MAX_SEED = 999999
+pipe = RAG_FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
 pipe = pipe.to("cuda")
 global run_nums