Update README.md
Browse files
README.md
CHANGED
|
@@ -21,7 +21,7 @@ from fofpred.schedulers.scheduling_flow_match_euler_discrete import FlowMatchEul
|
|
| 21 |
from PIL import Image
|
| 22 |
|
| 23 |
pipeline = FOFPredPipeline.from_pretrained(
|
| 24 |
-
"
|
| 25 |
torch_dtype=torch.bfloat16,
|
| 26 |
).to("cuda")
|
| 27 |
|
|
|
|
| 21 |
from PIL import Image
|
| 22 |
|
| 23 |
pipeline = FOFPredPipeline.from_pretrained(
|
| 24 |
+
"Salesforce/FOFPred",
|
| 25 |
torch_dtype=torch.bfloat16,
|
| 26 |
).to("cuda")
|
| 27 |
|