Hugging Face's logo

Imagroune
/
feynmodel

Safetensors
FeynModel
custom_code
Model card Files
xet
Community
Imagroune commited on
Commit
3554dc6
·
verified ·
1 Parent(s): 10d2316

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -0
README.md CHANGED
@@ -14,10 +14,12 @@ license: apache-2.0
14
  # how to use
15
 
16
  ```python
 
17
  from transformers import AutoProcessor, AutoModelForCausalLM
18
  model_id='Imagroune/feynmodel'
19
  processor = AutoProcessor.from_pretrained(model_id, trust_remote_code=True)
20
  model = AutoModelForCausalLM.from_pretrained(model_id,trust_remote_code=True)
 
21
  model.to('cuda')
22
  ```
23
 
 
14
  # how to use
15
 
16
  ```python
17
+ # make sur to have torch, transformers, pillow, einos ,einops and timm libraries
18
  from transformers import AutoProcessor, AutoModelForCausalLM
19
  model_id='Imagroune/feynmodel'
20
  processor = AutoProcessor.from_pretrained(model_id, trust_remote_code=True)
21
  model = AutoModelForCausalLM.from_pretrained(model_id,trust_remote_code=True)
22
+ # if have a cuda device
23
  model.to('cuda')
24
  ```
25