Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -51,12 +51,12 @@ GitHub repository rather than a stock `diffusers` pipeline.
|
|
| 51 |
import torch
|
| 52 |
from huggingface_hub import snapshot_download
|
| 53 |
|
| 54 |
-
from models.flux2 import Flux2,
|
| 55 |
from inference import load_state_dict
|
| 56 |
|
| 57 |
path = snapshot_download("JeonghyeokDo/GeoCore-9B")
|
| 58 |
|
| 59 |
-
model = Flux2(
|
| 60 |
model.load_state_dict(load_state_dict(path), strict=True)
|
| 61 |
model.eval()
|
| 62 |
```
|
|
|
|
| 51 |
import torch
|
| 52 |
from huggingface_hub import snapshot_download
|
| 53 |
|
| 54 |
+
from models.flux2 import Flux2, GeoCore9BParams
|
| 55 |
from inference import load_state_dict
|
| 56 |
|
| 57 |
path = snapshot_download("JeonghyeokDo/GeoCore-9B")
|
| 58 |
|
| 59 |
+
model = Flux2(GeoCore9BParams()).to("cuda", torch.bfloat16)
|
| 60 |
model.load_state_dict(load_state_dict(path), strict=True)
|
| 61 |
model.eval()
|
| 62 |
```
|