JeonghyeokDo commited on
Commit
a284772
·
verified ·
1 Parent(s): d386935

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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, TerraNova9BParams
55
  from inference import load_state_dict
56
 
57
  path = snapshot_download("JeonghyeokDo/GeoCore-9B")
58
 
59
- model = Flux2(TerraNova9BParams()).to("cuda", torch.bfloat16)
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
  ```