AhmedBenElsayedBenAhmed commited on
Commit
7e48196
·
1 Parent(s): e13652c

Merge branch 'Ai' into main

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -1,6 +1,10 @@
1
  # ===============================
2
  # 1️⃣ استيراد المكتبات
3
  # ===============================
 
 
 
 
4
  from transformers import AutoModelForCausalLM, AutoTokenizer, Trainer, TrainingArguments
5
  from peft import LoraConfig, get_peft_model, TaskType, PeftModel
6
  from flask import Flask, request, jsonify
 
1
  # ===============================
2
  # 1️⃣ استيراد المكتبات
3
  # ===============================
4
+ import os
5
+ os.environ["HF_HOME"] = "/tmp"
6
+ os.environ["TRANSFORMERS_CACHE"] = "/tmp"
7
+ os.environ["PEFT_HOME"] = "/tmp"
8
  from transformers import AutoModelForCausalLM, AutoTokenizer, Trainer, TrainingArguments
9
  from peft import LoraConfig, get_peft_model, TaskType, PeftModel
10
  from flask import Flask, request, jsonify