juancopi81 commited on
Commit
714a252
Β·
1 Parent(s): 74d5c4c

Remove fp16, using CPU

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -30,8 +30,8 @@ pipe = DiffusionPipeline.from_pretrained(
30
  detection_pipeline=language_detection_pipeline,
31
  translation_model=trans_model,
32
  translation_tokenizer=trans_tokenizer,
33
- revision="fp16",
34
- torch_dtype=torch.float16,
35
  )
36
 
37
  pipe.enable_attention_slicing()
 
30
  detection_pipeline=language_detection_pipeline,
31
  translation_model=trans_model,
32
  translation_tokenizer=trans_tokenizer,
33
+ # revision="fp16",
34
+ # torch_dtype=torch.float16,
35
  )
36
 
37
  pipe.enable_attention_slicing()