Spaces:

NJU
/

RAG-Diffusion

Running on Zero

znchen commited on 11 days ago

Commit

793fa41

•

1 Parent(s): a97032a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,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