palondomus commited on
Commit
f8eb8df
·
1 Parent(s): de83a49

CaesarFrenchLLM first test

Browse files
Files changed (1) hide show
  1. caesarfrenchllm.py +1 -1
caesarfrenchllm.py CHANGED
@@ -1,6 +1,6 @@
1
  import torch
2
  import os
3
- os.environ['TRANSFORMERS_CACHE'] = "./.cache"
4
  from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig, TextStreamer
5
  from vigogne.preprocess import generate_inference_chat_prompt
6
  class CaesarFrenchLLM:
 
1
  import torch
2
  import os
3
+ #os.environ['TRANSFORMERS_CACHE'] = "./.cache"
4
  from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig, TextStreamer
5
  from vigogne.preprocess import generate_inference_chat_prompt
6
  class CaesarFrenchLLM: