Spaces:

dahara1
/

FanFic-Illustrator

Running

dahara1 commited on Mar 21

Commit

29e82b9

verified ·

1 Parent(s): 255a49e

Upload prompt_generator.py

Files changed (1) hide show

prompt_generator.py CHANGED Viewed

@@ -119,8 +119,8 @@ def load_model():
         _model = AutoModelForCausalLM.from_pretrained(
             model_path,
             torch_dtype=torch_dtype,
-            device_map=device_map,
-            use_cache=True,
             low_cpu_mem_usage=True,
         )

         _model = AutoModelForCausalLM.from_pretrained(
             model_path,
             torch_dtype=torch_dtype,
+#            device_map=device_map,
+#            use_cache=True,
             low_cpu_mem_usage=True,
         )