ddwkim commited on
Commit
7c444f5
·
verified ·
1 Parent(s): 5f9f670

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -3
README.md CHANGED
@@ -85,13 +85,12 @@ from transformers.dynamic_module_utils import get_class_from_dynamic_module
85
 
86
  MODEL_ID = "KRAFTON/Raon-Speech-9B-FP8"
87
 
88
- _cfg = AutoConfig.from_pretrained(MODEL_ID, trust_remote_code=True)
89
  RaonPipeline = get_class_from_dynamic_module(
90
  "modeling_raon.RaonPipeline",
91
  MODEL_ID,
92
- revision=getattr(_cfg, "_commit_hash", None),
93
  )
94
- del _cfg
95
 
96
  pipe = RaonPipeline(MODEL_ID, device="cuda", dtype="bfloat16")
97
  ```
 
85
 
86
  MODEL_ID = "KRAFTON/Raon-Speech-9B-FP8"
87
 
88
+ config = AutoConfig.from_pretrained(MODEL_ID, trust_remote_code=True)
89
  RaonPipeline = get_class_from_dynamic_module(
90
  "modeling_raon.RaonPipeline",
91
  MODEL_ID,
92
+ revision=getattr(config, "_commit_hash", None),
93
  )
 
94
 
95
  pipe = RaonPipeline(MODEL_ID, device="cuda", dtype="bfloat16")
96
  ```