methodya commited on
Commit
0e4a5b6
·
verified ·
1 Parent(s): bb64f0f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -1,11 +1,12 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- # استخدام النموذج مباشرة بدون LoRA
5
  pipe = pipeline(
6
  "text-generation",
7
  model="deepseek-ai/deepseek-coder-1.3b-base",
8
- device_map="auto"
 
9
  )
10
 
11
  def generate_response(prompt):
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ # تعديل إعدادات pipeline
5
  pipe = pipeline(
6
  "text-generation",
7
  model="deepseek-ai/deepseek-coder-1.3b-base",
8
+ device_map="cpu", # نحدد CPU بشكل صريح
9
+ low_cpu_mem_usage=False # نعطل هذا الخيار
10
  )
11
 
12
  def generate_response(prompt):