Moore-Language-Space-ZeroGPU

Sleeping

ArissBandoss commited on May 18

Commit

eda40b7

verified ·

1 Parent(s): 9d96957

Update goai_helpers/goai_traduction.py

Files changed (1) hide show

goai_helpers/goai_traduction.py CHANGED Viewed

@@ -1,6 +1,7 @@
 import torch
 import spaces
 from transformers import pipeline, AutoModelForSeq2SeqLM, AutoTokenizer
 import os
 from huggingface_hub import login

 import torch
 import spaces
 from transformers import pipeline, AutoModelForSeq2SeqLM, AutoTokenizer
+from peft import PeftModel, PeftConfig
 import os
 from huggingface_hub import login