Spaces:

akhaliq
/

SRPO

Runtime error

akhaliq HF Staff commited on Sep 13

Commit

1a974b2

verified ·

1 Parent(s): 8bac254

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ from safetensors.torch import load_file
 # Load the model
 pipe = FluxPipeline.from_pretrained(
-    './data/flux',
     torch_dtype=torch.bfloat16,
     use_safetensors=True
 ).to('cuda')

 # Load the model
 pipe = FluxPipeline.from_pretrained(
+    'black-forest-labs/FLUX.1-dev',
     torch_dtype=torch.bfloat16,
     use_safetensors=True
 ).to('cuda')