Update inference-cpp docker (#12882)
* remove nouse run.py * add WORKDIR /llm
This commit is contained in:
		
							parent
							
								
									3f6ecce508
								
							
						
					
					
						commit
						4f2f92afa3
					
				
					 2 changed files with 2 additions and 3 deletions
				
			
		| 
						 | 
					@ -91,3 +91,5 @@ RUN set -eux && \
 | 
				
			||||||
    # Clean up unnecessary dependencies to reduce image size
 | 
					    # Clean up unnecessary dependencies to reduce image size
 | 
				
			||||||
    find /usr/lib/python3/dist-packages/ -name 'blinker*' -exec rm -rf {} + && \
 | 
					    find /usr/lib/python3/dist-packages/ -name 'blinker*' -exec rm -rf {} + && \
 | 
				
			||||||
    rm -rf /root/.cache/Cypress
 | 
					    rm -rf /root/.cache/Cypress
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					WORKDIR /llm/
 | 
				
			||||||
| 
						 | 
					
 | 
				
			||||||
| 
						 | 
					@ -6,13 +6,10 @@ init-llama-cpp
 | 
				
			||||||
# change the model_path to run
 | 
					# change the model_path to run
 | 
				
			||||||
if [[ "$DEVICE" == "Arc" || "$DEVICE" == "ARC" ]]; then
 | 
					if [[ "$DEVICE" == "Arc" || "$DEVICE" == "ARC" ]]; then
 | 
				
			||||||
    source ipex-llm-init -g --device Arc
 | 
					    source ipex-llm-init -g --device Arc
 | 
				
			||||||
    python run.py
 | 
					 | 
				
			||||||
elif [[ "$DEVICE" == "Flex" || "$DEVICE" == "FLEX" ]]; then
 | 
					elif [[ "$DEVICE" == "Flex" || "$DEVICE" == "FLEX" ]]; then
 | 
				
			||||||
    source ipex-llm-init -g --device Flex
 | 
					    source ipex-llm-init -g --device Flex
 | 
				
			||||||
    python run.py
 | 
					 | 
				
			||||||
elif [[ "$DEVICE" == "Max" || "$DEVICE" == "MAX" ]]; then
 | 
					elif [[ "$DEVICE" == "Max" || "$DEVICE" == "MAX" ]]; then
 | 
				
			||||||
    source ipex-llm-init -g --device Max
 | 
					    source ipex-llm-init -g --device Max
 | 
				
			||||||
    python run.py
 | 
					 | 
				
			||||||
else
 | 
					else
 | 
				
			||||||
    echo "Invalid DEVICE specified."
 | 
					    echo "Invalid DEVICE specified."
 | 
				
			||||||
fi
 | 
					fi
 | 
				
			||||||
| 
						 | 
					
 | 
				
			||||||
		Loading…
	
		Reference in a new issue