Spaces:

Remostartdev
/

FARMLINGUA_AI_CONVERSATIONAL

Running on CPU Upgrade

drrobot9 commited on Jan 28

Commit

5423133

verified ·

1 Parent(s): 1712d51

Update app/agents/crew_pipeline.py

Files changed (1) hide show

app/agents/crew_pipeline.py CHANGED Viewed

@@ -80,8 +80,7 @@ translation_model = AutoModelForSeq2SeqLM.from_pretrained(
 )
-if DEVICE == 'cpu':
-    translation_model = translation_model.to('cpu')