Hugging Face's logo

Spaces:
gustavoaq
/
llama_7b_ft_2
Paused

App Files Community
5
gustavoaq commited on
Commit
4d9a92b
·
1 Parent(s): 499f12f

Update finetune.py

Browse files
Files changed (1) hide show
  1. finetune.py +0 -1
finetune.py CHANGED
@@ -5,7 +5,6 @@ import pickle
5
  import random
6
  import json
7
  import torch.nn as nn
8
- import bitsandbytes as bnb
9
  from datasets import load_dataset
10
  import transformers
11
  from transformers import LlamaForCausalLM, LlamaTokenizer
 
5
  import random
6
  import json
7
  import torch.nn as nn
 
8
  from datasets import load_dataset
9
  import transformers
10
  from transformers import LlamaForCausalLM, LlamaTokenizer