Spaces:

pseudolab
/

KOMUChat

Paused

4n3mone commited on Nov 8, 2023

Commit

8a03b34

•

1 Parent(s): 717cb6d

Update koalpaca.py

Files changed (1) hide show

koalpaca.py CHANGED Viewed

@@ -2,6 +2,7 @@ import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig, GenerationConfig
 from peft import PeftModel, PeftConfig
 from model import Model
 class KoAlpaca(Model):
     def __init__(self):

 from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig, GenerationConfig
 from peft import PeftModel, PeftConfig
 from model import Model
+from accelerate import Accelerator
 class KoAlpaca(Model):
     def __init__(self):