Point quickstart at new 9B nonzeroshot merge
Browse files
README.md
CHANGED
|
@@ -34,7 +34,7 @@ dtype = torch.bfloat16
|
|
| 34 |
|
| 35 |
transformer = Flux2Transformer2DModel.from_pretrained(
|
| 36 |
"FunAILab/Open-Vision-Banana",
|
| 37 |
-
subfolder="final_models/9B/nonzeroshot/checkpoint-
|
| 38 |
torch_dtype=dtype,
|
| 39 |
)
|
| 40 |
pipe = Flux2KleinPipeline.from_pretrained(
|
|
|
|
| 34 |
|
| 35 |
transformer = Flux2Transformer2DModel.from_pretrained(
|
| 36 |
"FunAILab/Open-Vision-Banana",
|
| 37 |
+
subfolder="final_models/9B/nonzeroshot/checkpoint-avg82500-85000-87500-92500-97500_uniform/transformer",
|
| 38 |
torch_dtype=dtype,
|
| 39 |
)
|
| 40 |
pipe = Flux2KleinPipeline.from_pretrained(
|